diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py index ef71889def18d2de891937a63f6c0658f27f061e..1ea73262ba0a7771633b9e93531e7fc3257ba778 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: