A deep learning model using TensorFlow and Keras to predict stock prices based on historical data.
- Uses LSTMs or other deep learning models for prediction
- Processes historical stock market data
- Visualizes trends and predictions
- Can be trained on custom datasets
📦 stock-prediction-ai
┣ 📂 data # Stores datasets
┣ 📂 models # Trained models
┣ 📂 notebooks # Jupyter notebooks for experiments
┣ 📂 scripts # Python scripts for training & evaluation
┣ 📜 train.py # Main training script
┣ 📜 predict.py # Script for making predictions
┣ 📜 requirements.txt # List of dependencies
┗ 📜 README.md # Project documentation
1️⃣ Clone this repository:
git clone https://github.com/Nxtyyaaaa/stock-prediction-ai.git
cd stock-prediction-ai 2️⃣ Install dependencies:
pip install -r requirements.txt 3️⃣ Run training:
python train.py 4️⃣ Make predictions:
python predict.py - Algorithm: LSTM / GRU / CNN
- Dataset: Yahoo Finance / Alpha Vantage / Custom CSV
- Loss Function: MSE / RMSE
- Optimizer: Adam / SGD
- Graphical representation of stock trends
- Comparison of predicted vs actual prices
- Adding more technical indicators
- Using Transformer-based models
- Building a real-time prediction API
Developed by Nitya Matta ✨
MIT License – Feel free to use and modify!