The final project for machine learning.
Switch branches/tags
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.


Machine Learning Final Project

Find out more here Alt text

Github repository for the machine learning trading algorithm. Still a work in progress. The goal is to use reinforcement learning in order to determine when to buy or hold a stock.


  1. Have pip installed on your system
  2. Run pip install -r requirements.txt, make sure this is for python3, may need to specify pip3
  3. Go to the pytorch website and install for your OS
  4. Run the command jupyter lab inside src directory
  5. Go into the ipython notebook for reinforcement learning and play around

Running Code

The main class that contains the data for processing in this code is financial data. This class imports 500 different stocks, cleans and scales the data. After this process is performed reinforcment learning is used in order to train to buy, sell or hold stocks. This code can be seen int he RL ipython notebook.

jupyter lab

Northeastern University 2018