From 0d92d7ece5141b641c922a9d6defb08f24f298fc Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Tue, 6 Dec 2022 00:24:39 +0100
Subject: [PATCH] wip
---
convlab/policy/emoTUS/evaluate.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 579315f7..4f7f6d28 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -219,9 +219,9 @@ class Evaluator:
result[metric] = sum(scores[metric])/len(scores[metric])
print(f"{metric}: {result[metric]}")
emo_score = emotion_score(golden_emotions, gen_emotions)
- for metric in emo_score:
- result[metric] = emo_score[metric]
- print(f"{metric}: {result[metric]}")
+ # for metric in emo_score:
+ # result[metric] = emo_score[metric]
+ # print(f"{metric}: {result[metric]}")
result["dialog"] = dialog_result
--
GitLab