Skip to content

Using Reinforcement Learning with Deep Deterministic Policy Gradient for Portfolio Optimization

Notifications You must be signed in to change notification settings

jnenx/ddpg-rl-portfolio-management

 
 

Repository files navigation

Using Reinforcement Learning for Portfolio Optimization

Algorithms used in this work

  • Long short term memory
  • Deep Deterministic Policy Gradient

Dataset

  • S&P500 dataset from kaggle found here

References

Getting Started

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.

Prerequisites

To install the required python packges, browse to the code folder then run pip install --user --requirement requirements.txt

Running the tests

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.

Running the training

To train the model from scratch and overwrite the saved weights, run stock_trading.py. This could take several hours.

License

This project is licensed under the MIT License.

About

Using Reinforcement Learning with Deep Deterministic Policy Gradient for Portfolio Optimization

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • HTML 52.6%
  • Jupyter Notebook 45.8%
  • Python 1.6%