添加人工操作,修改环境bug
This commit is contained in:
parent
db890f83cf
commit
64935bf92f
2
env.py
2
env.py
@ -124,7 +124,7 @@ class PartitionMazeEnv(gym.Env):
|
||||
for i in range(len(self.row_cuts) - 1):
|
||||
for j in range(len(self.col_cuts) - 1):
|
||||
d = (self.col_cuts[j+1] - self.col_cuts[j]) * self.W * \
|
||||
(self.row_cuts[i] + self.row_cuts[i+1]) * self.H
|
||||
(self.row_cuts[i+1] - self.row_cuts[i]) * self.H
|
||||
rho_time_limit = (self.flight_time_factor - self.trans_time_factor) / \
|
||||
(self.comp_time_factor - self.trans_time_factor)
|
||||
rho_energy_limit = (self.battery_energy_capacity - self.flight_energy_factor * d - self.trans_energy_factor * d) / \
|
||||
|
15
human_action.py
Normal file
15
human_action.py
Normal file
@ -0,0 +1,15 @@
|
||||
from env import PartitionMazeEnv
|
||||
|
||||
env = PartitionMazeEnv()
|
||||
|
||||
state = env.reset()
|
||||
print(state)
|
||||
|
||||
action_series = [[0], [0], [0.5], [0], [0.1], [0.3]]
|
||||
|
||||
for i in range(10):
|
||||
action = action_series[i]
|
||||
state, reward, done, info, _ = env.step(action)
|
||||
print(state, reward, done, info)
|
||||
if done:
|
||||
break
|
Loading…
Reference in New Issue
Block a user