以下内容来源于郭宪老师《深入浅出强化学习》,总结记录:
如何将建好的环境进行注册,以便通过gym的标准形式进行调用:
- 将我们自己的环境文件(我创建的文件名为grid_mdp.py)拷贝到gym安装目录E:\Anaconda\envs\paddle_env\Lib\site-packages\gym\envs\classic_control文件夹中
- 打开该文件夹(第一步中的classic_control文件夹)下的__init__.py文件,在文件末尾加入语句:
from gym.envs.classic_control.grid_mdp import GridEnv
- 进入你的gym安装目录E:\Anaconda\envs\paddle_env\Lib\site-packages\gym\envs,打开该文件夹下的__init__.py文件,添加代码:
register(
id='GridWorld-v0',
entry_point='gym.envs.classic_control:GridEnv',
max_episode_steps=200,
reward_threshold=100.0,
)
第一个参数‘id’就是你调用gym.make(‘id’)时的id,这个id你可以随便选取,这里的名字是GridWorld-v0
第二个参数就是函数路口了。后面的参数原则上来说可以不必要写。经过以上三步,就完成了注册。