From 58da110ba2a9f2068889d9b1c60098eec5f74c65 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Tue, 6 Dec 2022 00:01:56 +0100
Subject: [PATCH] wip
---
convlab/policy/emoTUS/evaluate.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 715553fa..a8a59da8 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -235,7 +235,7 @@ def emotion_score(golden_emotions, gen_emotions):
macro_f1 = metrics.f1_score(golden_emotions, gen_emotions, average="macro")
sep_f1 = metrics.f1_score(
golden_emotions, gen_emotions, average=None, labels=labels)
- cm = metrics.confusion_matrix(golden_emotions, gen_emotions, labels)
+ cm = metrics.confusion_matrix(golden_emotions, gen_emotions, labels=labels)
disp = metrics.ConfusionMatrixDisplay(
confusion_matrix=cm, display_labels=labels)
disp.plot()
--
GitLab