Stock Trading with Machine Learning

Overview

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

Requirements

• Python 3.5+
• alpha_vantage
• pandas
• numpy
• sklearn
• keras
• tensorflow
• matplotlib

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