如何注册强化学习测试使用gym仿真环境
如何注册强化学习测试使用gym仿真环境1、OpenAI Gym是训练强化学习时经常使用的仿真环境,利用python的gym库就可以使用,安装:pip install gym2、注册环境,有时训练模型时需要自己的环境,可以自行注册。代码结构如图,其中mygym为自己写的环境,包括初始化、reset、step。RLTest为测试文件,查看环境能否使用。3、环境封装,在__init__.py文件中写如下
·
如何注册强化学习测试使用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、测试使用。
更多推荐


所有评论(0)