YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
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:
- Downloads last month
- 81