Skip to content

Instantly share code, notes, and snippets.

@pyliaorachel

pyliaorachel/cartpole_env_extract.py

Created June 15, 2018 05:11
Show Gist options
  • Save pyliaorachel/26affd3b6cfbf68e3b2e408b91401358 to your computer and use it in GitHub Desktop.
Save pyliaorachel/26affd3b6cfbf68e3b2e408b91401358 to your computer and use it in GitHub Desktop.
Download ZIP
OpenAI Gym CartPole - Deep Q-Learning (cartpole env)
Raw
cartpole_env_extract.py
...
if not done:
reward = 1.0
elif self.steps_beyond_done is None:
# Pole just fell!
self.steps_beyond_done = 0
reward = 1.0
else:
self.steps_beyond_done += 1
reward = 0.0
...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment