Skip to content

q-learning-trader/q-trading-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 88.1%
  • Python 11.9%