如何注册强化学习测试使用gym仿真环境

1、OpenAI Gym是训练强化学习时经常使用的仿真环境,利用python的gym库就可以使用,安装:pip install gym
2、注册环境,有时训练模型时需要自己的环境,可以自行注册。代码结构如图,其中mygym为自己写的环境,包括初始化、reset、stepRLTest为测试文件,查看环境能否使用。
在这里插入图片描述
3、环境封装,在__init__.py文件中写如下代码。注明封装的环境名称,所在位置。

from gym.envs.registration import register
register(
    id='Simulator-v0',
    entry_point='mygym.SimulatorEnv:Simulator',#第一个myenv是文件夹名字,第二个myenv是文件名字,MyEnv是文件内类的名字
)

4、环境具体内容。按自己需要写。
在这里插入图片描述
5、测试使用。
在这里插入图片描述

更多推荐