- Long short term memory
- Deep Deterministic Policy Gradient
- S&P500 dataset from kaggle found here
- A Deep Reinforcement Learning Framework for the Financial Portfolio Management Problem
- Continuous control with deep reinforcement learning
- The code is inspired by CSCI 599 deep learning and its applications final project
- The environment is inspired by wassname/rl-portfolio-management
- DDPG implementation is inspired by Deep Deterministic Policy Gradients in TensorFlow
These instructions will get you a copy of the project up and running on your local machine for development and testing purposes. See deployment for notes on how to deploy the project on a live system.
To install the required python packges, browse to the code folder then run pip install --user --requirement requirements.txt
ddpg_tests.ipynb is a step by step jupyter notebook showing the performance of the trained agent on unseen stocks. You can run this jupyter notebook directly without having to run the training since the training weights are saved in the weigths folder.
To train the model from scratch and overwrite the saved weights, run stock_trading.py. This could take several hours.
This project is licensed under the MIT License.