|
|
---
|
|
|
library_name: tensorflow
|
|
|
tags:
|
|
|
- lstm
|
|
|
- stock-prediction
|
|
|
- time-series
|
|
|
- financial-analysis
|
|
|
- tensorflow
|
|
|
- keras
|
|
|
pipeline_tag: tabular-regression
|
|
|
---
|
|
|
|
|
|
# Universal LSTM Stock Prediction Model
|
|
|
|
|
|
## Model Description
|
|
|
|
|
|
This is a Universal LSTM model trained for stock price prediction with sentiment analysis integration. The model can predict stock prices for multiple stocks using technical indicators and sentiment features.
|
|
|
|
|
|
## Model Details
|
|
|
|
|
|
- **Model Type**: LSTM (Long Short-Term Memory) Neural Network
|
|
|
- **Framework**: TensorFlow/Keras
|
|
|
- **Task**: Time Series Prediction / Stock Price Forecasting
|
|
|
- **Training Date**: July 5, 2025
|
|
|
- **Input Features**: Technical indicators + Sentiment scores
|
|
|
- **Output**: Next-day stock price prediction
|
|
|
|
|
|
## Architecture
|
|
|
|
|
|
- **Input Shape**: (None, 60, 6)
|
|
|
- **Output Shape**: (None, 1)
|
|
|
- **Parameters**: 31,651
|
|
|
|
|
|
The model uses a multi-layer LSTM architecture optimized for financial time series prediction.
|
|
|
|
|
|
## Training Data
|
|
|
|
|
|
- **Stocks**: Multiple stocks including AAPL, MSFT, GOOGL, AMZN, TSLA, META, NVDA
|
|
|
- **Features**:
|
|
|
- Technical indicators (Open, High, Low, Close, Volume)
|
|
|
- Sentiment scores from financial news analysis
|
|
|
- Moving averages and other derived features
|
|
|
- **Time Period**: Historical stock data with corresponding news sentiment
|
|
|
- **Preprocessing**: StandardScaler normalization
|
|
|
|
|
|
## Performance
|
|
|
|
|
|
The model has been evaluated on multiple stocks with varying performance metrics. Best performance achieved:
|
|
|
- **RMSE**: Varies by stock (typically 5-15 points)
|
|
|
- **Architecture**: Optimized through hyperparameter tuning
|
|
|
|
|
|
## Usage
|
|
|
|
|
|
```python
|
|
|
import tensorflow as tf
|
|
|
import pickle
|
|
|
import numpy as np
|
|
|
|
|
|
# Load the model
|
|
|
model = tf.keras.models.load_model('stage2_universal_lstm_20250705_170829.keras')
|
|
|
|
|
|
# Load the scaler
|
|
|
with open('stage2_scalers_20250705_170829.pkl', 'rb') as f:
|
|
|
scalers = pickle.load(f)
|
|
|
|
|
|
# Prepare your data (X should be scaled using the same scaler)
|
|
|
# X = your_data # Shape: (batch_size, sequence_length, features)
|
|
|
# predictions = model.predict(X)
|
|
|
```
|
|
|
|
|
|
## Files Included
|
|
|
|
|
|
- `stage2_universal_lstm_20250705_170829.keras`: Main model file
|
|
|
- `stage2_scalers_20250705_170829.pkl`: Feature scalers for preprocessing
|
|
|
- `stage2_metadata_20250705_170829.json`: Model metadata and configuration
|
|
|
- `stage2_architecture_20250705_170829.txt`: Detailed architecture description
|
|
|
|
|
|
## Citation
|
|
|
|
|
|
If you use this model in your research, please cite:
|
|
|
|
|
|
```
|
|
|
@misc{lstm_stock_prediction_2025,
|
|
|
title={Universal LSTM Stock Prediction Model with Sentiment Analysis},
|
|
|
author={Jeng Yang},
|
|
|
year={2025},
|
|
|
url={https://huggingface.co/jengyang/lstm-stock-prediction-model}
|
|
|
}
|
|
|
```
|
|
|
|
|
|
## Disclaimer
|
|
|
|
|
|
This model is for research and educational purposes only. Stock market predictions are inherently uncertain and this model should not be used as the sole basis for financial decisions. Always consult with financial professionals and conduct your own research before making investment decisions.
|
|
|
|
|
|
## Contact
|
|
|
|
|
|
For questions or issues, please open an issue in the model repository.
|
|
|
|
