DRL-for-Energy-Systems-Optimal-Scheduling/training_info_plot_MIP_with_different_sigma.py at main · Pumpkin772/DRL-for-Energy-Systems-Optimal-Scheduling · GitHub

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
47
48
49
50
import pickle
import torch
import pandas as pd
from plotDRL import plot_training_rewardinfo, plot_evaluation_information, plot_cost_rewardinfo, plot_cost_rewardinfo_MIP, plot_training_unbalanceinfo
import matplotlib.pyplot as plt
datasource1 = 'D:\桌面\\test\MIP-DQN\AgentMIPDQN\\all_seeds_reward_record_sigma20.pkl'
name1 = 'simga20'
color1 = 'blue'
datasource2 = 'D:\桌面\\test\MIP-DQN\AgentMIPDQN\\all_seeds_reward_record_sigma50.pkl'
name2 = 'sigma50'
color2 = 'green'
datasource3 = 'D:\桌面\\test\MIP-DQN\AgentMIPDQN\\all_seeds_reward_record_sigma100.pkl'
name3 = 'sigma100'
color3 = 'pink'

plot_training_rewardinfo(name1,datasource1,color1)
plot_training_rewardinfo(name2,datasource2,color2)
plot_training_rewardinfo(name3,datasource3,color3)
plt.legend()
# 添加标题和轴标签
plt.title('Training rewards over Episodes with 95% Confidence Interval')
plt.xlabel('Episode')
plt.ylabel('Reward')
# 添加虚线网格
plt.grid(True, linestyle='--', linewidth=0.5, color='gray', alpha=0.7)
plt.show()

plot_cost_rewardinfo(name1,datasource1,color1)
plot_cost_rewardinfo(name2,datasource2,color2)
plot_cost_rewardinfo(name3,datasource3,color3)
plt.legend()
# 添加标题和轴标签
plt.title('Training Cost over Episodes with 95% Confidence Interval')
plt.xlabel('Episode')
plt.ylabel('Cost')
# 添加虚线网格
plt.grid(True, linestyle='--', linewidth=0.5, color='gray', alpha=0.7)
plt.show()

plot_training_unbalanceinfo(name1,datasource1,color1)
plot_training_unbalanceinfo(name2,datasource2,color2)
plot_training_unbalanceinfo(name3,datasource3,color3)
plt.legend()
# 添加标题和轴标签
plt.title('Training unbalance over Episodes with 95% Confidence Interval')
plt.xlabel('Episode')
plt.ylabel('Unbalance')
# 添加虚线网格
plt.grid(True, linestyle='--', linewidth=0.5, color='gray', alpha=0.7)
plt.show()