Fixed shape of observation

This commit is contained in:
Iaroslav Shcherbatyi
2016-05-31 21:34:44 +02:00
parent 1635a46c7f
commit d5bfb848bb

View File

@@ -48,7 +48,7 @@ class CNNClassifierTraining(gym.Env):
# observation features, in order: num of instances, num of labels,
# validation accuracy after training with given parameters
self.observation_space = spaces.Box(-1e5, 1e5, 3) # validation accuracy
self.observation_space = spaces.Box(-1e5, 1e5, 2) # validation accuracy
# Start the first game
self._reset()