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