添加人工操作,修改环境bug

This commit is contained in:
weixin_46229132 2025-03-14 09:42:56 +08:00
parent db890f83cf
commit 64935bf92f
2 changed files with 16 additions and 1 deletions

2
env.py
View File

@ -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
View 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