tianying107 / q-trading-pytorch

DQN stock trading pytorch implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Q-Trader

** Use in your own risk **

Pytorch implmentation from q-trader(https://github.com/edwardhdlu/q-trader)

Results

Some examples of results on test sets:

HSI2018
Starting Capital: $100,000.
HSI, 2017-2018. Profit of $10702.13.

Running the Code

To train the model, download a training and test csv files from Yahoo! Finance into data/

mkdir models
python train ^GSPC 10 1000

Then when training finishes (minimum 200 episodes for results):

jupyter notebook -> visualize.ipynb

References

Deep Q-Learning with Keras and Gym - Q-learning overview and Agent skeleton code

About

DQN stock trading pytorch implementation


Languages

Language:Jupyter Notebook 88.1%Language:Python 11.9%