添加人工操作,修改环境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 i in range(len(self.row_cuts) - 1):
|
||||||
for j in range(len(self.col_cuts) - 1):
|
for j in range(len(self.col_cuts) - 1):
|
||||||
d = (self.col_cuts[j+1] - self.col_cuts[j]) * self.W * \
|
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) / \
|
rho_time_limit = (self.flight_time_factor - self.trans_time_factor) / \
|
||||||
(self.comp_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) / \
|
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