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