Downloads · 30 days
215
17% of all-time downloads
suryaR-15/lstm-stock-price-predictor
lstm-stock-price-predictor is a machine learning model from suryaR-15. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This is a custom LSTM model trained on the Google stock price prediction dataset from Kaggle.
Downloads · 30 days
215
17% of all-time downloads
All-time downloads
1.2K
Public
Repo size
2 MB
Likes
13
Trending 1
Click a slice to open those files.
.bin608 KB · 100%
From the Hugging Face model README
This is a custom LSTM model trained on the Google stock price prediction dataset from Kaggle.
The following hyperparameters were used during training:
'input_dim': 1, 'hidden_dim': 64, 'num_layers': 5, 'output_dim': 1, 'num_epochs': 500
Training loss curve:

Predicted vs. actual prices on the training dataset:

Predicted vs. actual prices on the test dataset:
