diff --git a/requirements.txt b/requirements.txt index fd78bad..eee88c1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,6 @@ appdirs==1.4.2 -APScheduler==3.3.1 -atari-py==0.0.18 -box2d-py==2.3.1 cycler==0.10.0 -gym==0.8.0 imageio==2.1.2 -Keras==1.2.2 -matplotlib==2.0.0 mujoco-py==0.5.7 numpy==1.12.0 olefile==0.44 @@ -28,6 +22,5 @@ six==1.10.0 sklearn==0.0 tensorflow==1.0.0 termcolor==1.1.0 -Theano==0.8.2 tqdm==4.11.2 tzlocal==1.3 diff --git a/train/RLBrain.py b/train/RLBrain.py index ebfd3bd..3951fcf 100644 --- a/train/RLBrain.py +++ b/train/RLBrain.py @@ -83,9 +83,6 @@ def subTrain(self, isTrain, dates): # Q learning start action = self.chooseAction(state, isTrain) - if not isTrain: - print(action) - reward = 0 if action == STOP: reward = account.stop()