From 7bc4f213c09752ad511c20e724971602be14b671 Mon Sep 17 00:00:00 2001 From: Hsien-Chin Lin <linh@hhu.de> Date: Mon, 20 Feb 2023 17:51:35 +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 ef71889d..1ea73262 100644 --- a/convlab/policy/emoTUS/evaluate.py +++ b/convlab/policy/emoTUS/evaluate.py @@ -259,7 +259,7 @@ class Evaluator: for metric_type, score in scores.items(): result[metric_type] = {} for m, s in score.items(): - result[metric_type][m] = sum(s[m])/len(s[m]) + result[metric_type][m] = sum(s)/len(s) print(f"{metric_type}-{m}: {result[metric_type][m]}") if not golden_emotion: -- GitLab