This repository contains the code for our CS 337 project - Deep Ensemble Reinforcement Learning for Adaptive Trading. The team members are: Shubham Hazra (210100143), Om Godage (21d100006) and Vijay Balsubramaniam (21d180043). The project focuses on the application of Deep Reinforcement Learning for trading and explores the use of ensemble methods to improve the performance of the agent. We also compare the performance of the agent with the baseline index fund and statistical algorithms like HRP and CVaR under Modern Portfolio Theory.
The code is written in Python 3.10. The required packages are listed in requirements.txt
.
To install the packages, run the following command:
pip install -r requirements.txt
./
: Contains the report and the requirements file.src
: Contains the source code for the project.results
: Contains the results of the experiments.plots
: Contains the plots of the results.
The code files are present in the src
directory.
The three main files are:
ensemble.ipynb
: Contains the code for the ensemble methods.MPT.py
: Contains the code for the statistical algorithms.plot.ipynb
: Contains the code to plot the results.
You can just run the cells in the notebook to reproduce the results. The results are stored in the results
directory and the plots are stored in the plots
directory.