gym是常用的强化学习包,下面是几个入门的程序
查看用到的游戏
from gym import envs
print(envs.registry.all())
一个立杆子游戏
import gym
env = gym.make('CartPole-v0')
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()