里面如何创建新的module_如何在Gym中创建一个新的环境

68d468dd1f0269e4896698489f21573c.png

创建一个名为 gym-foo 的环境(这也是一个PIP包)

创建的环境至少包含以下几项内容:

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-foo包之后,我们可以用 gym.make('gym_foo:foo-v0') 创建环境的实例。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值