q-trading-pytorch
q-trading-pytorch copied to clipboard
DQN stock trading pytorch implementation
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:
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