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

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

1、OpenAI Gym是训练强化学习时经常使用的仿真环境，利用python的gym库就可以使用，安装：pip install gym
2、注册环境，有时训练模型时需要自己的环境，可以自行注册。代码结构如图，其中mygym为自己写的环境，包括初始化、reset、step。RLTest为测试文件，查看环境能否使用。

3、环境封装，在__init__.py文件中写如下代码。注明封装的环境名称，所在位置。

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

4、环境具体内容。按自己需要写。

5、测试使用。