OpenAI Gym--创建新环境

这是目录

• 如何为Gym创建新环境

  • 创建一个名为gym-foo的新存储库，它也应该是一个PIP包。

  • 一个很好的例子是https://github.com/openai/gym-soccer。

  • 它至少应具有以下文件：

    gym-foo/
      README.md
      setup.py
      gym_foo/
        __init__.py
        envs/
          __init__.py
          foo_env.py
          foo_extrahard_env.py
    

  • gym-foo/setup.py 应该有：

    from setuptools import setup
    
    setup(name='gym_foo',
          version='0.0.1',
          install_requires=['gym']  # And any other dependencies foo needs
    )
    

  • gym-foo/gym_foo/__init__.py 应该有：

    from gym.envs.registration import register
    
    register(
        id='foo-v0',
        entry_point='gym_foo.envs:FooEnv',
    )
    register(
        id='foo-extrahard-v0',
        entry_point='gym_foo.envs:FooExtraHardEnv',
    )
    

  • gym-foo/gym_foo/envs/__init__.py 应该有：

    from gym_foo.envs.foo_env import FooEnv
    from gym_foo.envs.foo_extrahard_env import FooExtraHardEnv
    

  • gym-foo/gym_foo/envs/foo_env.py 应该看起来像：

    import gym
    from gym import error, spaces, utils
    from gym.utils import seeding
    
    class FooEnv(gym.Env):
      metadata = {'render.modes': ['human']}
    
      def __init__(self):
        ...
      def step(self, action):
        ...
      def reset(self):
        ...
      def render(self, mode='human'):
        ...
      def close(self):
        ...
    

  • 使用安装pip install -e gym-foo完软件包后，您可以使用创建环境的实例gym.make('gym_foo:foo-v0')