【发布时间】:2021-01-27 20:29:31
【问题描述】:
我正在尝试为我的强化学习算法创建一个环境,但是,在调用 PPOPolicy 时似乎有点问题。为此我开发了以下环境envFru:
import gym
import os, sys
import numpy as np
import pandas as pd
from gym import spaces
import random
class envFru(gym.Env):
metadata ={'render.modes': ['human']}
def __init__(self):
self.df = df
self.action_space = spaces.Discrete(2)
self.observation_space = spaces.Box(low=np.array([0,0,0]), high=np.array([1,1,1]), dtype=np.float16)
def reset(self):
pass
def step(self, action):
pass
def _next_observation(self):
pass
def _take_action(self, action):
pass
def render(self, mode = 'human', close=False):
pass
from stable_baselines.common.vec_env import DummyVecEnv
from stable_baselines.common.policies import MlpPolicy
from stable_baselines2.ppo.ppo import PPO
envF = DummyVecEnv([lambda : envFru()])
model = PPOPolicy(envF, MlpPolicy, learning_rate= 0.001)
model.learn(total_timesteps=20000)
obs = env.reset()
for i in range(MAX_EPISODES):
action, _states = model.predict(obs)
obs, reward,done,info = env.step(action)
#env.render()
我得到的回溯如下:
AttributeError Traceback (most recent call last)
<ipython-input-124-550b8c75c26b> in <module>
12 envF = DummyVecEnv([lambda : envFruit()])
13
---> 14 model = PPOPolicy(envF, MlpPolicy, learning_rate= 0.001)
15 model.learn(total_timesteps=20000)
16
~\Desktop\ImitationLearning\stable_baselines2\ppo\policies.py in __init__(self, observation_space, action_space, learning_rate, net_arch, activation_fn, adam_epsilon, ortho_init, log_std_init)
29 ortho_init=True, log_std_init=0.0):
30 super(PPOPolicy, self).__init__(observation_space, action_space)
---> 31 self.obs_dim = self.observation_space.shape[0]
32
33 # Default network architecture, from stable-baselines
AttributeError: 'DummyVecEnv' object has no attribute 'shape'
【问题讨论】:
标签: python reinforcement-learning openai-gym