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