RL:环境注册

  • Post author:
  • Post category:其他


以下内容来源于郭宪老师《深入浅出强化学习》,总结记录:

如何将建好的环境进行注册,以便通过gym的标准形式进行调用:

  1. 将我们自己的环境文件(我创建的文件名为grid_mdp.py)拷贝到gym安装目录E:\Anaconda\envs\paddle_env\Lib\site-packages\gym\envs\classic_control文件夹中
  2. 打开该文件夹(第一步中的classic_control文件夹)下的__init__.py文件,在文件末尾加入语句:
from gym.envs.classic_control.grid_mdp import GridEnv
  1. 进入你的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

第二个参数就是函数路口了。后面的参数原则上来说可以不必要写。经过以上三步,就完成了注册。