File size: 1,075 Bytes
a283aa6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import json
from env.ev_charge_env import EVChargeEnv
from agent.baseline_agent import BaselineAgent
def run_episode(env, agent):
obs, _ = env.reset()
total_reward = 0.0
steps = 0
while True:
action = agent.select_action(obs)
obs, reward, terminated, truncated, _ = env.step(action)
total_reward += reward
steps += 1
if terminated or truncated or steps >= 200:
break
return total_reward, steps
def main():
env = EVChargeEnv()
agent = BaselineAgent()
rewards = []
steps_list = []
for _ in range(5):
total_reward, steps = run_episode(env, agent)
rewards.append(total_reward)
steps_list.append(steps)
output = {
"avg_reward": sum(rewards) / len(rewards),
"avg_steps": sum(steps_list) / len(steps_list),
"episodes": len(rewards)
}
print(json.dumps(output))
# Save JSON for reproducibility
with open("sample_output.json", "w") as f:
json.dump(output, f, indent=4)
if __name__ == "__main__":
main()
|