Update grace_eval.py

grace_eval.py

@@ -1,30 +1,38 @@
+import matplotlib
+matplotlib.use('Agg')
 import matplotlib.pyplot as plt
+import numpy as np
+from PIL import Image
+import io
+
+DIMENSIONS = ["Generalization", "Relevance", "Artistry", "Efficiency"]
 
 def compute_sample_scores(results, prompt):
-    # 示例评分 - 实际应用中可以用更复杂的评估逻辑
+    """模拟评分 - 实际项目应替换为真实评估逻辑"""
     return {
-        "sd_v1_5": {"G": 4, "R": 4, "A": 4, "E": 3},
-        "openjourney_v4": {"G": 3, "R": 4, "A": 5, "E": 3},
-        "ldm_256": {"G": 2, "R": 3, "A": 3, "E": 5}
+        "sd_v1_5": [4, 4, 4, 3],
+        "openjourney_v4": [3, 4, 5, 3],
+        "ldm_256": [2, 3, 3, 5]
     }
 
 def plot_radar(scores_dict, out_path="radar.png"):
-    categories = list(scores_dict.values())[0].keys()
-    N = len(categories)
+    fig = plt.figure(figsize=(8, 8))
+    ax = fig.add_subplot(111, polar=True)
     
-    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
+    angles = np.linspace(0, 2*np.pi, len(DIMENSIONS), endpoint=False)
+    angles = np.concatenate((angles, [angles[0]]))
     
     for model, scores in scores_dict.items():
-        values = list(scores.values())
-        values += values[:1]  # 闭合雷达图
-        angles = [n / float(N) * 2 * 3.14159 for n in range(N)]
-        angles += angles[:1]
-        ax.plot(angles, values, linewidth=1, linestyle='solid', label=model)
-        ax.fill(angles, values, alpha=0.1)
+        data = np.concatenate((scores, [scores[0]]))
+        ax.plot(angles, data, linewidth=2, linestyle='solid', label=model)
+        ax.fill(angles, data, alpha=0.1)
     
-    ax.set_xticks(angles[:-1])
-    ax.set_xticklabels(categories)
-    ax.set_title("GRACE 评估雷达图 (CPU模式)", size=12, y=1.1)
+    ax.set_thetagrids(angles[:-1] * 180/np.pi, DIMENSIONS)
+    ax.set_title("GRACE Evaluation Radar", size=14, pad=20)
     ax.legend(loc='upper right')
-    plt.savefig(out_path, bbox_inches='tight')
-    plt.close()
+    
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png', bbox_inches='tight')
+    plt.close()
+    buf.seek(0)
+    return Image.open(buf)