Python gym_space_spec Exemples

Exemple #1

Fichier : gym_spaces_utils_test.py Projet : Alwin-Bucher/tensor2tensor-1

 def test_discrete_space_spec(self):
     discrete_space = Discrete(100)
     spec = gym_spaces_utils.gym_space_spec(discrete_space)
     self.assertIsInstance(spec, tf.FixedLenFeature)
     self.assertEqual(spec.dtype, tf.int64)
     self.assertListEqual(list(spec.shape), [1])

Exemple #2

Fichier : gym_spaces_utils_test.py Projet : Alwin-Bucher/tensor2tensor-1

 def test_box_space_spec(self):
     box_space = Box(low=0, high=10, shape=[5, 6], dtype=np.float32)
     spec = gym_spaces_utils.gym_space_spec(box_space)
     self.assertIsInstance(spec, tf.FixedLenFeature)
     self.assertEqual(spec.dtype, tf.float32)
     self.assertListEqual(list(spec.shape), [5, 6])

Exemple #3

Fichier : env_problem.py Projet : induszing/tensor2tensor

 def action_spec(self):
   """The spec for reading an observation stored in a tf.Example."""
   return gym_spaces_utils.gym_space_spec(self.action_space)