使用OpenAI自定义环境

Question

我有一个文件rb.py，其结构大致如下：

import gym

def main():
   env = gym.make('ComplexMiniGolf-v0')

问题在于，如果我创建另一个文件并执行以下操作：

from rb import main
main()

自定义环境已加载，一切正常。但如果相反：

import gym
env = gym.make('ComplexMiniGolf-v0')

即使加载了完全相同的程序包，也无法找到自定义环境。

Answer 1

您是否已在健身房注册环境？

如果您的环境在以下结构中定义，则可以这样做：

myenv/
    __init__.py
    myenv.py

myenv.py包含您的环境的类。在__init__.py中，输入以下代码：

from gym.envs.registration import register

register(
    id='MyEnv-v0',
    entry_point='myenv.myenv:MyEnv',
)

然后您可以通过以下方式使用它：

import gym
import myenv
env = gym.make('MyEnv-v0')