强化学习环境:Gym本地库中添加自己编写的环境

本文详细介绍了如何在Gym库中创建并注册自定义环境,包括定位gym库、编辑环境文件、修改__init__.py及注册环境等步骤,以便在个人项目中使用。

1. 找到本地gym库地址下的envs目录,可能是:python3.6/site-packages/gym/envs。

    envs下面有多个目录,选择一个你的环境所属的目录(不是很重要,选择相近类别即可)

2. 假如我们选择的是classic_control目录,进入该目录。

3. 把自己编写的环境文件(比如myenv.py)拷贝至该目录。

4. 打开该目录下的__init__.py文件,加入如下语句:

from gym.envs.classic_control.myenv import LargeGridWorld

from gym.envs.classic_control.myenv import SimpleGridWorld

from gym.envs.classic_control.myenv import WindyGridWorld

5. 返回至envs目录下,打开该目录下的__init__.py文件,加入如下语句完成环境的注册:

register(

id='LargeGridWorld-v0',

entry_point='gym.envs.classic_control:LargeGridWorld',

max_episode_steps=999,

)
register(

id='SimpleGridWorld-v0',

entry_point='gym.envs.classic_control:SimpleGridWorld',

max_episode_steps=999,

)
register(

id='WindyGridWorld-v0',

entry_point='gym.envs.classic_control:WindyGridWorld',

max_episode_steps=999,

)

其中每一个注册方法内的id好像一定要带上版本号,也就是id后半部分的”-v0”不能省。

6. 完成了上述工作后,在自己的代码里就可以直接使用下面的语句来构建相应的环境对象了:

id = "SimpleGridWorld-v0"	# 设置环境id

env = gym.make(id)	# 构建环境对象

env.reset()	# 重置环境

env.render()	# 绘制环境

Input("press any key to continue...")

env.close()	# 关闭环境

 

 

 

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值