Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/enviroments.py

Created December 23, 2019 10:40
Show Gist options
  • Save NMZivkovic/dd8277b2df070713051b98a2e90d913b to your computer and use it in GitHub Desktop.
Save NMZivkovic/dd8277b2df070713051b98a2e90d913b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
enviroments.py
train_env = suite_gym.load('CartPole-v0')
evaluation_env = suite_gym.load('CartPole-v0')
print('Observation Spec:')
print(train_env.time_step_spec().observation)
print('Reward Spec:')
print(train_env.time_step_spec().reward)
print('Action Spec:')
print(train_env.action_spec())
train_env = tf_py_environment.TFPyEnvironment(train_env)
evaluation_env = tf_py_environment.TFPyEnvironment(evaluation_env)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment