如何注册强化学习测试使用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、测试使用。