From 81324d3da72ac25f27951b36b18d2c5cb7d9ee20 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Mon, 20 Feb 2023 17:42:32 +0100
Subject: [PATCH] wip

---
 convlab/policy/emoTUS/evaluate.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 3fbd348a..ef71889d 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -247,11 +247,11 @@ class Evaluator:
         # full action
         for gen_act, golden_act in zip(gen_acts, golden_acts):
             s = f1_measure(preds=gen_act, labels=golden_act)
-            for metric in scores:
+            for metric in scores["complete"]:
                 scores["complete"][metric].append(s[metric])
             s = f1_measure(preds=self._intent_domain(gen_act),
                            labels=self._intent_domain(golden_act))
-            for metric in scores:
+            for metric in scores["intent_domain"]:
                 scores["intent_domain"][metric].append(s[metric])
 
         result = {}
-- 
GitLab