-
如何为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')
-
OpenAI Gym--创建新环境
最新推荐文章于 2024-06-12 16:39:33 发布