Skip to content

This project is a Stock Trader trained to trade stocks from the S&P 500. It was made using a Deep Q-Learning model and libraries such as TensorFlow, Keras, and OpenAI Gym. It was trained on data from 2006-2016, cross validated on data from 2016-2018, and tested on data from 2018-2021

Notifications You must be signed in to change notification settings

rishipython/StockTradingAI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 

Repository files navigation

StockTradingAI

This project is a Stock Trader trained to trade stocks from the S&P 500. It was made using a Deep Q-Learning model and libraries such as TensorFlow, Keras, and OpenAI Gym. It was trained on data from 2006-2016, cross validated on data from 2016-2018, and tested on data from 2018-2021.

Installation

All of the code is in Google Colab Notebook. Simply download it and upload it into your Google Drive.

Cross Validation Results

image

image

image

image

image

Based on the data shown above, model 3 was determined to be the best.

Test Results

image

image

About

This project is a Stock Trader trained to trade stocks from the S&P 500. It was made using a Deep Q-Learning model and libraries such as TensorFlow, Keras, and OpenAI Gym. It was trained on data from 2006-2016, cross validated on data from 2016-2018, and tested on data from 2018-2021

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published