gym环境注册

荣荣moon

已于 2022-03-06 19:04:32 修改

阅读量1.6k

点赞数 2

文章标签： gym环境 classic_control 自定义env register entry_point

于 2022-03-06 18:23:17 首次发布

本文链接：https://blog.csdn.net/qq_43413627/article/details/123314396

版权

1.将我们自己的环境文件拷贝到你的gym安装目录/gym/gym/envs/classic_control文件夹中。（拷贝在这个文件夹中因为要使用rendering模块。

附上我自己服务器的目录（因为用了虚拟环境所以与上文不太相同）：

/data/jy/anaconda3/envs/py/lib/python3.6/site-packages/gym/envs/classic_control/

结果如下图：

2.打开该文件夹（第一步中的文件夹）下的__init__.py文件，在文件末尾加入语句：

from gym.envs.classic_control.comb import combEnv
from gym.envs.classic_control.sweeping import sweepingEnv
from gym.envs.classic_control.dynamic import dynamicEnv
from gym.envs.classic_control.intelligent import intelligentEnv

完整版如图

3.进入文件夹你的gym安装目录/gym/gym/envs，

打开该文件夹下的__init__.py文件，添加代码：

register(
    id="comb-v0",
    entry_point="gym.envs.classic_control:combEnv",
)

register(
    id="sweeping-v0",
    entry_point="gym.envs.classic_control:sweepingEnv",
)

register(
    id="dynamic-v0",
    entry_point="gym.envs.classic_control:dynamicEnv",
)

register(
    id="intelligent-v0",
    entry_point="gym.envs.classic_control:intelligentEnv",
)

完成。

参考：

https://zhuanlan.zhihu.com/p/26985029