/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

Primary LanguageJupyter Notebook

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