Added transfer learning to make predictinos and generating response f…#5
Merged
ArshdeepSingh728281 merged 1 commit intoOpen-Source-Chandigarh:mainfrom Oct 23, 2025
Conversation
ArshdeepSingh728281
merged commit 9e52ee1
into
Open-Source-Chandigarh:main
1 check failed
ArshdeepSingh728281
added
hacktoberfest-accepted
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Transfer Learning Implementation
Summary
Reduces prediction time from 2-5 minutes to 4 seconds (75x speedup) using transfer learning.
Changes
New Files:
pythonModel/Train_Base_Model.py- Trains base model on 9 diverse stockspythonModel/models/.gitignore- Excludes weight filespythonModel/models/README.md- Setup documentationModified:
pythonModel/pipline.py- Loads pre-trained weights, freezes LSTM layers, reduces epochs from 30 to 5Performance
Training Output
Expand logs
Setup
cd pythonModel
python Train_Base_Model.py
Creates
base_stock_weights.weights.h5. Automatically used when present, falls back to full training if missing.Testing
curl "http://localhost:8000/predict?ticker=INFY.NS&symbol=INFY&period=1mo&interval=1h&days_to_fetch=3"
Returns valid JSON in 4-10 seconds.
Compatibility