quantylab / rltrader

파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)
https://contents.premium.naver.com/misoncorp/quantylab
MIT License
349 stars 357 forks source link

colab에서 구현할 수 있나요? #100

Open boozesounds opened 3 years ago

boozesounds commented 3 years ago

제가 코랩을 이용하는데, 코랩에서도 구현 가능한가요..? 파일이 많이 나누어져있어서 어렵네요..

ggokka commented 3 years ago

tensorflow 1.15.2 버전을 써야 하는데요. colab에서는 다운버전을 어떻게 하는지 모르겠네요.

2021-06-13 01:39:03.204078: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0 Traceback (most recent call last): File "main.py", line 66, in from learners import ReinforcementLearner, DQNLearner, \ File "/content/drive/My Drive/quantylab/rltrader/learners.py", line 11, in from networks import Network, DNN, LSTMNetwork, CNN File "/content/drive/My Drive/quantylab/rltrader/networks.py", line 23, in from tensorflow.keras.backend import set_session ImportError: cannot import name 'set_session' from 'tensorflow.keras.backend' (/usr/local/lib/python3.7/dist-packages/tensorflow/keras/backend/init.py)

이슈71번 참고 https://github.com/quantylab/rltrader/issues/71

ggokka commented 3 years ago

아, 데탑하고 똑같이 하면 되네요. 현 버전이 언인스톨되면서 다운그레이드 됩니다. !pip install tensorflow==1.15.2

파라메터는 86번 이슈보고 따라 했습니다 https://github.com/quantylab/rltrader/issues/86 python main.py --stock_code 005930 005380 015760 --rl_method a3c --net lstm --num_steps 5 --learning --num_epoches 1000 --lr 0.001 --start_epsilon 1 --discount_factor 0.9 --output_name train --start_date 20170101 --end_date 20181231