This project focuses on predicting the stock prices of The State Bank Of India using machine learning regression algorithms. The dataset was collected from Yahoo Finance and contains historical stock data.
- Utilizes various regression algorithms for stock price prediction.
- Dataset collected from Yahoo Finance for The State Bank Of India.
We implemented the following regression algorithms for stock price prediction:
| 🤖 Algorithm | 📜 Description |
|---|---|
| Linear Regression | A basic regression algorithm. |
| Support Vector Regression (SVR) | Effective for non-linear relationships. |
| Random Forest | Ensemble learning method using decision trees. |
| Gradient Boosting Models (GBM) | Sequentially builds models to improve predictions. |
| Extreme Gradient Boosting (XGBoost) | Advanced boosting technique with regularization. |
| AdaBoostRegressor | Combines multiple weak learners. |
| Decision Tree | Simple yet effective model based on tree structure. |
| KNeighborsRegressor (KNN) | Predicts based on nearest neighbors' average. |
| Artificial Neural Networks (ANN) | Mimics human brain for complex data patterns. |
| Long Short Term Memory (LSTM) | Suitable for time-series prediction. |
The dataset used in this project is sourced from Yahoo Finance and includes historical stock data for The State Bank Of India. It comprises relevant features such as:
- 📈 Open prices
- 📉 High prices
- 📉 Low prices
- 💵 Close prices
- 📦 Volume
📂 data/ # Contains the dataset files. 📓 notebooks/ # Jupyter notebooks with the code for data exploration, preprocessing, and model training. 🐍 src/ # Python source code for the project. 📋 requirements.txt # List of dependencies needed to run the project.
- Install dependencies using:
pip install -r requirements.txt
- Execute the notebooks in the
notebooks/folder in the given order. - Run the scripts in the
src/folder for further analysis or model training.
The sequence of all the algorithms used is as follows:
- Linear Regression
- SVR
- Random Forest
- Gradient Boosting Models (GBM)
- Extreme Gradient Boosting (XGBoost)
- AdaBoostRegressor
- Decision Tree
- KNeighborsRegressor (KNN)
- Artificial Neural Networks (ANN)
- Long Short Term Memory (LSTM)
The Root Mean Square Error (RMSE) of all the following 10 Regression Algorithms is provided below:
The Mean Absolute Error (MAE) of all the following 10 Regression Algorithms is provided below:
The Mean Absolute Percentage Error (MAPE) of all the following 10 Regression Algorithms is provided below:
- Combine this data with stock sentiment data to enhance prediction accuracy.
- Utilize clustering algorithms to develop a buy/sell recommendation system.
Among the models assessed, AdaBoostRegressor and LSTM emerged as the top performers, showcasing low RMSE, MAE, and MAPE values. These metrics suggest that these algorithms effectively capture the underlying trends and patterns in the stock price data, making them reliable for prediction tasks.
While some models demonstrated solid predictive capabilities, others, such as Support Vector Regression (SVR) and KNeighborsRegressor, recorded higher RMSE and MAE values. This indicates that these algorithms may yield acceptable predictions on average but are susceptible to significant errors in certain scenarios, emphasizing the need for careful model selection for stock price predictions.
Rohit Dubey 👨💻
We welcome contributions to this project! Please see our Contributing.md file for guidelines on how to get involved.
This project is licensed under the MIT License.