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