gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-20 05:52:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes to CEM:

Browse Source 
* Take gym environment as command line argument

* Call seed

* Fix reference to RandomAgent -> CEM
This commit is contained in:
Jie Tang
2016-06-21 15:06:31 -07:00
parent 1e86e73919
commit e5be288d3a
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/agents/cem.py
View File

@@ -52,10 +52,12 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--display', action='store_true') parser.add_argument('--display', action='store_true')
parser.add_argument('target', nargs="?", default="CartPole-v0")
args = parser.parse_args() args = parser.parse_args()
env = gym.make(args.target)
env.seed(0)
np.random.seed(0) np.random.seed(0)
env = gym.make('CartPole-v0')
params = dict(n_iter=10, batch_size=25, elite_frac = 0.2) params = dict(n_iter=10, batch_size=25, elite_frac = 0.2)
num_steps = 200 num_steps = 200
@@ -94,5 +96,5 @@ if __name__ == '__main__':
env.monitor.close() env.monitor.close()
logger.info("Successfully ran RandomAgent. Now trying to upload results to the scoreboard. If it breaks, you can always just try re-uploading the same results.") logger.info("Successfully ran cross-entropy method. Now trying to upload results to the scoreboard. If it breaks, you can always just try re-uploading the same results.")
gym.upload(outdir) gym.upload(outdir)