From 214ffc29a87eabb546a7d7dccc7a3e487cc5995f Mon Sep 17 00:00:00 2001 From: Hsien-Chin Lin <linh@hhu.de> Date: Mon, 17 Oct 2022 14:50:06 +0200 Subject: [PATCH] wip --- convlab/policy/genTUS/evaluate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/convlab/policy/genTUS/evaluate.py b/convlab/policy/genTUS/evaluate.py index 82779afe..87de8549 100644 --- a/convlab/policy/genTUS/evaluate.py +++ b/convlab/policy/genTUS/evaluate.py @@ -182,9 +182,9 @@ class Evaluator: dialog_result.append(d) else: gen_acts, golden_acts = [], [] - for gen_act, golden_act in zip(gen_file['dialog']["gen_acts"], gen_file['dialog']["golden_acts"]): - gen_acts.append(usr_action) - golden_acts.append(ans_action) + for dialog in gen_file['dialog']: + gen_acts.append(dialog["gen_acts"]) + golden_acts.append(dialog["golden_acts"]) dialog_result = gen_file['dialog'] scores = {"precision": [], "recall": [], "f1": [], "turn_acc": []} -- GitLab