rishipython / 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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Jupyter Notebook 100.0%