Home

Awesome

Stock Trading with Machine Learning

Overview

A stock trading bot that uses machine learning to make price predictions.

Requirements

Documentation

Blog Post

Medium Article

Train your own model

  1. Clone the repo
  2. Pip install the requirements pip install -r requirements.txt
  3. Save the stock price history to a csv file python save_data_to_csv.py --help
  4. Edit one of the model files to accept the symbol you want
  5. Edit model architecture
  6. Edit dataset preprocessing / history_points inside util.py
  7. Train the model python tech_ind_model.py or python basic_model.py
  8. Try the trading algorithm on the newly saved model python trading_algo.py

License

GPL-3.0