强化学习记录——gym中各种环境信息查看


一、gym中所有可以用的模拟环境

from gym import envs
for env in envs.registry.all():
    print(env.id)

二、环境展示

import gym
env = gym.make('Pendulum-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()

三、其他环境参数

import gym
env = gym.make('Pendulum-v0').unwrapped
print(env.action_space)  # 输出动作信息
#print(env.action_space.n)  # 输出动作个数
print(env.observation_space)  # 查看状态空间
print(env.observation_space.shape[0])  # 输出状态个数
print(env.observation_space.high)  # 查看状态的最高值
print(env.observation_space.low)  # 查看状态的最低值

四、各类环境

经典控制环境Classic control:入门

算法学习环境Algorithms

2D仿真环境Box2D

Mujoco环境

Atari

文本游戏环境Toy text

五、官网提供

gym入门、环境鸟瞰图:Gym

环境脚本:gym/gym/envs/classic_control at master · openai/gym · GitHub

  • 8
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
强化学习是一种通过智能体与环境的交互来学习最优行为的方法。Gym一个用于开发和比较强化学习算法的工具包,它提供了一系列标准化的环境,使得研究者可以更加方便地进行算法的实现和比较。 要编写一个Gym环境,需要实现一个Python类,该类必须继承gym.Env类,并实现以下方法: 1. __init__(self): 初始化环境,包括环境状态的初始化、动作空间的定义等。 2. reset(self): 重置环境状态,返回初始状态。 3. step(self, action): 执行一个动作,返回下一个状态、奖励和是否终止的标志。 4. render(self, mode='human'): 可选方法,用于可视化环境状态。 下面是一个简单的Gym环境示例,该环境一个简单的迷宫游戏,智能体需要找到宝藏并躲避陷阱: ```python import gym from gym import spaces import numpy as np class MazeEnv(gym.Env): def __init__(self): self.action_space = spaces.Discrete(4) # 动作空间为上下左右四个方向 self.observation_space = spaces.Box(low=0, high=255, shape=(5, 5, 3), dtype=np.uint8) # 状态空间为一个5x5的RGB图像 self.maze = np.array([ [0, 0, 0, 0, 0], [0, 1, 1, 0, 0], [0, 0, 1, 1, 0], [0, 1, 1, 0, 0], [0, 0, 0, 0, 2] ]) # 迷宫地图,0表示空地,1表示陷阱,2表示宝藏 self.agent_pos = np.array([1, 1]) # 智能体初始位置 self.reward_range = (-1, 1) # 奖励范围为[-1, 1] def reset(self): self.agent_pos = np.array([1, 1]) # 重置智能体位置 return self._get_obs() def step(self, action): if action == 0: # 上 self.agent_pos[0] = max(0, self.agent_pos[0] - 1) elif action == 1: # 下 self.agent_pos[0] = min(4, self.agent_pos[0] + 1) elif action == 2: # 左 self.agent_pos[1] = max(0, self.agent_pos[1] - 1) elif action == 3: # 右 self.agent_pos[1] = min(4, self.agent_pos[1] + 1) reward = self._get_reward() done = self._is_done() obs = self._get_obs() return obs, reward, done, {} def render(self, mode='human'): pass def _get_obs(self): obs = np.zeros((5, 5, 3), dtype=np.uint8) obs[self.agent_pos[0], self.agent_pos[1], :] = [255, 255, 255] # 智能体位置为白色 obs[self.maze == 1, 0] = 255 # 陷阱位置为红色 obs[self.maze == 2, 2] = 255 # 宝藏位置为蓝色 return obs def _get_reward(self): if self.maze[tuple(self.agent_pos)] == 1: # 踩到陷阱 return -1 elif self.maze[tuple(self.agent_pos)] == 2: # 找到宝藏 return 1 else: return 0 def _is_done(self): return self.maze[tuple(self.agent_pos)] != 0 ``` 这个环境比较简单,状态空间为一个5x5的RGB图像,动作空间为上下左右四个方向,奖励为-1(踩到陷阱)、0(未找到宝藏)和1(找到宝藏)。在实际应用,需要根据具体问题设计合适的状态空间、动作空间和奖励函数。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值