Skip to content

Enhancing Stock Market Prediction with Guided Stochastic Gradient Descent

Notifications You must be signed in to change notification settings

ECOLS-research-group/GRNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Stock Market Prediction with Guided Stochastic Gradient Descent (GSGD)
This project explores the application of Guided Stochastic Gradient Descent (GSGD) with Recurrent Neural Networks (RNNs) to enhance the performance of stock price prediction models. The study compares the performance of GSGD-optimized RNNs (GRNN) with standard RNN and Long Short-Term Memory (LSTM) networks.

Overview
Stock market prediction is a challenging task due to the complex and volatile nature of financial markets. This project uses machine learning techniques, particularly RNNs, to model sequential data for predicting stock price movements. The project investigates the use of GSGD, an advanced optimization technique, to improve the performance of RNNs in handling inconsistent data and revisiting batches based on loss.

Installation
 Clone the repository:
   git clone https://github.com/ECOLS-research-group/GRNN.git
   cd stock-market-prediction
 Create a virtual environment:
   python3 -m venv venv
   source venv/bin/activate # On Windows use venv\Scripts\activate
 Install dependencies:
   pip install -r requirements.txt

Usage
 Prepare the data:
   Ensure your dataset is placed in the data/ directory.
   Modify data_preparation.py to point to your dataset and preprocess it as needed.
Train the models:
   Run the main script to train and evaluate the models:
   python main.py

Training and Evaluation
The project includes training scripts for three types of models:
 Standard RNN (train_rnn.py)
 LSTM (train_lstm.py)
 GRNN with GSGD (train_gsgd.py)
The main.py script handles the training and evaluation of these models, including the calculation of metrics such as accuracy, F1 score, and ROC-AUC.

Results
The results of the study are summarized in the results/ directory. Key metrics include accuracy, F1 score, and ROC-AUC for each model.

About

Enhancing Stock Market Prediction with Guided Stochastic Gradient Descent

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages