diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py index 579315f76f36ffa9c77fda8e0a2b05bc30c75ec8..4f7f6d28f502adf13e8cb0e14747bf35bfb4a424 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