在自定义的OpenAI体育馆环境中传递参数

Question

我对定制的OpenAI体育馆环境有疑问。在我的实现中，我进行了以下初始化：

 def __init__(self, df):

这意味着我在调用gym.make()时需要传递一个额外的参数（数据帧）。我读到存在两种不同的解决方案：第一种解决方案是在创建环境时修改register函数，第二种解决方案是在自定义环境中创建一个额外的初始化方法并对其进行访问以传递额外的方法。论点。所提出的解决方案中没有一个适合我。有人可以帮我吗？

Answer 1

您可以将__init__()方法更改为：

    class CustomEnv(gym.Env):    
      def __init__(self, df):
        super(CustomEnv, self).__init__()
        # you are ready to use your df

并像这样使用您的环境：

env = gym.make('CustomEnv-v0', df=df)