This repository is dedicated to building and improving a robust stock price prediction model using LSTM and other advanced machine learning techniques. The project aims to handle multiple stocks, extract meaningful features, and enhance prediction accuracy with continuous iteration and innovation.
I am constantly working on refining the model, expanding support for a broader range of stocks, and integrating advanced data-driven approaches to deliver more precise predictions. Contributions, suggestions, and collaborations are highly encouraged to make this project even better.
- Stock price prediction using LSTM.
- Data pre-processing for financial datasets.
- Model evaluation with metrics like RMSE and accuracy.
- Visualization of stock trends.
- Prediction of future stock prices for specific dates.
- Clone the repository
git clone https://github.com/tkim602/financeModeling.git cd financeModeling
- Set up a Python virtual environment
python3 -m venv stock_env source stock_env/bin/activate
- Install the dependencies
pip install -r requirements.txt
- Activate the virtual enviroment
source stock_env/bin/activate
- Run the main script
python main.py
- Predict stock price
- Ensure lstm_model_weights.weights.h5 and scaler.pkl are available in the project directory. These files are generated after running main.py.
- Run the prediction script
python predict.py
- Input the desired prediction date when prompted
Enter the date for prediction (YYYY-MM-DD): 2024-12-29
- The predicted stock price for the given date will be displayed
Predicted price for 2024-12-29: $129.32
- main.py: Script for training the LSTM model.
- predict.py: Script for predicting future stock prices.
- features.py: Contains feature engineering functions.
- indicators.py: Includes financial indicators like RSI, MACD, and Bollinger Bands.
- config.py: Configuration for dataset paths and model parameters.
- data_processing.py: Functions for loading and processing financial data.
- evaluate.py: Model evaluation metrics and visualization tools.