web-dev-qa-db-ja.com

JupyterノートブックのみにOpenAIジムを表示する

ノートブックでOpenAIジムをプレイしたいのですが、ジムはインラインでレンダリングされます。

基本的な例を次に示します。

import matplotlib.pyplot as plt
import gym
from IPython import display
%matplotlib inline

env = gym.make('CartPole-v0')
env.reset()

for i in range(25):
   plt.imshow(env.render(mode='rgb_array'))
   display.display(plt.gcf())    
   display.clear_output(wait=True)
   env.step(env.action_space.sample()) # take a random action

env.close()

これは機能し、ノートブックにジムが表示されます。

gym in notebook

しかし！また、まったく同じことを示す対話型ウィンドウも開きます。 このウィンドウを開きたくない：

interactive window

pythonpython-3.xjupyter-notebookreinforcement-learningopenai-gym
9
MasterScrat

ここでフォークできる実用的な例を作成しました： https://kyso.io/eoin/openai-gym-jupyter レンダリングの2つの例Jupyter-1つはmp4として、もう1つはリアルタイムgifとして。

.mp4の例は非常に単純です。

import gym
from gym import wrappers

env = gym.make('SpaceInvaders-v0')
env = wrappers.Monitor(env, "./gym-results", force=True)
env.reset()
for _ in range(1000):
    action = env.action_space.sample()
    observation, reward, done, info = env.step(action)
    if done: break
env.close()

その後、新しいセルで

import io
import base64
from IPython.display import HTML

video = io.open('./gym-results/openaigym.video.%s.video000000.mp4' % env.file_infix, 'r+b').read()
encoded = base64.b64encode(video)
HTML(data='''
    <video width="360" height="auto" alt="test" controls><source src="data:video/mp4;base64,{0}" type="video/mp4" /></video>'''
.format(encoded.decode('ascii')))
7
Eoin Murray