From d5bfb848bb38e3d44f53b07784918df5ceedf5dc Mon Sep 17 00:00:00 2001 From: Iaroslav Shcherbatyi Date: Tue, 31 May 2016 21:34:44 +0200 Subject: [PATCH] Fixed shape of observation --- gym/envs/parameter_tuning/train_deep_cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gym/envs/parameter_tuning/train_deep_cnn.py b/gym/envs/parameter_tuning/train_deep_cnn.py index a4ace3ece..48877ab58 100644 --- a/gym/envs/parameter_tuning/train_deep_cnn.py +++ b/gym/envs/parameter_tuning/train_deep_cnn.py @@ -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()