技術系メモ

雑多なメモです。ほんと自分用のwikiです。

【Open AI Gym】Open AI Gymの問題を解いた結果をuploadする方法

Open AI gymのcartpoleのサンプルコードに付け足すような形でメモしておく

以下、cartpoleをランダム行動により実行し、その結果をOpen AI gymにuploadするプログラム。

APIキーは自分のものを使用してください。

import gym
from gym import wrappers

env = gym.make('CartPole-v0')
outdir = './cartpole-experiment-0'  # 結果を出力するディレクトリpath
env = wrappers.Monitor(env, outdir)
for i_episode in range(20):
    observation = env.reset()
    for t in range(100):
        # env.render()
        print(observation)
        action = env.action_space.sample()
        observation, reward, done, info = env.step(action)
        if done:
            print("Episode finished after {} timesteps".format(t+1))
            break

env.close()
gym.upload(outdir, api_key='APIキー')