gym初步使用

$$gym初步使用$$

• 安装

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym

解决Pong-v0环境问题

pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py

GYM-OPENAI-环境平台

0.导入工具包

import gym
import random
import numpy as np
import matplotlib.pyplot as plt

1.激活environment , 查看基本信息

1) 激活环境

env = gym.make('Pong-v0')

2) 查看基本信息，包括state信息的shape,action space的大小

# state信息
env.observation_space

# state's shape
env.observation_space.shape

# action信息
env.action_space

# action个数
env.action_space.n

2.初始化环境，查看state具体信息

state = env.reset()
state

state.shape

3. 执行action并可视化效果

1) 执行action

state = env.reset()
env.render()

new_state,reward,done,info = env.step(0)   # 0 表示第一个Action
reward,done,info

new_state

new_state -state

plt.imshow(state)

plt.imshow(env.render(mode='rgb_array'))

2) 可视化Action

state = env.reset()
done =False
action =3
t = 0
for _ in range(2000):
    t += 1
    env.render()
    state,reward,done,_ = env.step(action)
    if done:
        break
t

4.关闭环境

env.close()