From 5bedcba1dd19deb37c1cb917d053ebaff9a94d0d Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Tue, 6 Dec 2022 00:19:14 +0100
Subject: [PATCH] wip

---
 convlab/policy/emoTUS/evaluate.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 9b479ca2..c5cd5fbe 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -241,7 +241,8 @@ def emotion_score(golden_emotions, gen_emotions):
         confusion_matrix=cm, display_labels=labels)
     disp.plot()
     plt.savefig("emotion.png")
-    r = {"macro_f1": macro_f1, "sep_f1": list(sep_f1), "cm": list(cm)}
+    r = {"macro_f1": macro_f1, "sep_f1": list(
+        sep_f1), "cm": [list(c) for c in list(cm)]}
     print(r)
     return r
 
-- 
GitLab