如何在Gym中创建自己的环境
-
首先,将写好的环境类(继承自
gym.env
)代码保存成为一个文件,假设名为class.py
。 -
进入
python
目录下的site-packages
中,找到文件夹gym/envs/classic_control
,将class.py
放入该目录下。 -
在当前目录下找到
__init__.py
并打开,在其中加入以下代码from gym.envs.classic_control.<your_file_name> import <your_environment_class_name>
。 -
回到父目录,找到
__init__.py
并打开,在文件中加入以下内容:register( id='your_env_name', entry_point='gym.envs.classic_control:<your_class_name>', max_episode_steps=200, # optional reward_threshold=100.0, # optional )