118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
|
import json
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
import numpy as np
|
|||
|
|
import tkinter as tk
|
|||
|
|
from tkinter import ttk
|
|||
|
|
from scipy.signal import savgol_filter
|
|||
|
|
|
|||
|
|
# 讀取多個 JSON 檔案
|
|||
|
|
def load_data(file_paths):
|
|||
|
|
all_data = []
|
|||
|
|
for file_path in file_paths:
|
|||
|
|
with open(file_path) as f:
|
|||
|
|
data = json.load(f)
|
|||
|
|
# 處理特殊的數據格式（numpy.float64 對象）
|
|||
|
|
processed_data = {}
|
|||
|
|
for key, value in data.items():
|
|||
|
|
if isinstance(value, list):
|
|||
|
|
# 檢查是否包含字典格式的數值
|
|||
|
|
if value and isinstance(value[0], dict) and 'value' in value[0]:
|
|||
|
|
processed_data[key] = [item['value'] for item in value]
|
|||
|
|
else:
|
|||
|
|
processed_data[key] = value
|
|||
|
|
else:
|
|||
|
|
processed_data[key] = value
|
|||
|
|
all_data.append(processed_data)
|
|||
|
|
return all_data
|
|||
|
|
|
|||
|
|
def smooth(y, window_length=51, polyorder=3):
|
|||
|
|
return savgol_filter(y, window_length, polyorder)
|
|||
|
|
|
|||
|
|
# 定義要繪製的數據
|
|||
|
|
def plot_data(data_list, keys, name_list, battle_name, smooth_window=2):
|
|||
|
|
for key in keys:
|
|||
|
|
# 創建一個圖形
|
|||
|
|
fig, ax = plt.subplots()
|
|||
|
|
# 設置坐標軸背景顏色
|
|||
|
|
ax.set_facecolor('lightyellow')
|
|||
|
|
# 設置網格顏色
|
|||
|
|
ax.grid(color='green', linestyle='--', linewidth=0.5)
|
|||
|
|
|
|||
|
|
for data, name in zip(data_list, name_list):
|
|||
|
|
if key in data:
|
|||
|
|
x = data[key + '_T']
|
|||
|
|
y = data[key]
|
|||
|
|
y_smooth = smooth(y)
|
|||
|
|
ax.plot(x, y_smooth, label=name)
|
|||
|
|
#ax.plot(data[key + '_T'], data[key], label=name)
|
|||
|
|
ax.set_xlabel('Time Steps')
|
|||
|
|
ax.set_ylabel(key)
|
|||
|
|
ax.set_title(battle_name)
|
|||
|
|
ax.legend()
|
|||
|
|
|
|||
|
|
plt.show()
|
|||
|
|
|
|||
|
|
def create_dynamic_window(data_list, keys, name_list, battle_name):
|
|||
|
|
root = tk.Tk()
|
|||
|
|
root.title("動態選擇要顯示的數據")
|
|||
|
|
|
|||
|
|
# 創建左側的選擇面板
|
|||
|
|
select_frame = ttk.Frame(root)
|
|||
|
|
select_frame.pack(side=tk.LEFT, fill=tk.Y, padx=10, pady=5)
|
|||
|
|
|
|||
|
|
# 創建變量來存儲選擇狀態
|
|||
|
|
vars = []
|
|||
|
|
|
|||
|
|
def update_plot(*args):
|
|||
|
|
# 獲取選中的數據
|
|||
|
|
selected_names = [name for name, var in zip(name_list, vars) if var.get()]
|
|||
|
|
filtered_data = [data for data, name in zip(data_list, name_list) if name in selected_names]
|
|||
|
|
filtered_names = [name for name in name_list if name in selected_names]
|
|||
|
|
|
|||
|
|
# 清除所有現有的圖表
|
|||
|
|
plt.close('all')
|
|||
|
|
# 重新繪製圖表
|
|||
|
|
if filtered_data: # 確保至少選擇了一個數據
|
|||
|
|
plot_data(filtered_data, keys, filtered_names, battle_name)
|
|||
|
|
|
|||
|
|
# 創建複選框
|
|||
|
|
for name in name_list:
|
|||
|
|
var = tk.BooleanVar(value=True) # 默認全選
|
|||
|
|
var.trace('w', update_plot) # 添加跟踪器，當值改變時更新圖表
|
|||
|
|
vars.append(var)
|
|||
|
|
cb = ttk.Checkbutton(select_frame, text=name, variable=var)
|
|||
|
|
cb.pack(anchor='w', padx=5, pady=2)
|
|||
|
|
|
|||
|
|
# 添加全選/取消全選按鈕
|
|||
|
|
def select_all():
|
|||
|
|
for var in vars:
|
|||
|
|
var.set(True)
|
|||
|
|
|
|||
|
|
def deselect_all():
|
|||
|
|
for var in vars:
|
|||
|
|
var.set(False)
|
|||
|
|
|
|||
|
|
ttk.Button(select_frame, text="全選", command=select_all).pack(pady=5)
|
|||
|
|
ttk.Button(select_frame, text="取消全選", command=deselect_all).pack(pady=5)
|
|||
|
|
|
|||
|
|
# 初始繪圖
|
|||
|
|
update_plot()
|
|||
|
|
|
|||
|
|
root.mainloop()
|
|||
|
|
|
|||
|
|
# 使用者選擇要繪製的數據
|
|||
|
|
file_paths = ['results/sacred/10gen_protoss/feudal/5/info.json',
|
|||
|
|
'results/sacred/10gen_protoss/qmix/6/info.json',
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
data_list = load_data(file_paths)
|
|||
|
|
selected_keys = ['test_battle_won_mean','return_mean',"worker_loss","loss_td","manager_loss"]
|
|||
|
|
#selected_keys = ['battle_won_mean', 'loss']
|
|||
|
|
name_list = ['feudal', 'qmix']
|
|||
|
|
|
|||
|
|
|
|||
|
|
battle_name = '5protoss'
|
|||
|
|
|
|||
|
|
# 使用新的動態選擇窗口
|
|||
|
|
create_dynamic_window(data_list, selected_keys, name_list, battle_name)
|