From 571d9ea0b4d48f704c88e28eb959e5085f8097fd Mon Sep 17 00:00:00 2001
From: linh <linh@hilbert314.hilbert.hpc.uni-duesseldorf.de>
Date: Tue, 18 Oct 2022 22:44:47 +0200
Subject: [PATCH] zim
---
convlab/evaluator/multiwoz_eval.py | 1 +
convlab/policy/genTUS/evaluate.py | 4 ----
convlab/policy/genTUS/stepGenTUS.py | 1 +
3 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/convlab/evaluator/multiwoz_eval.py b/convlab/evaluator/multiwoz_eval.py
index fa1cd0e4..9d02c2ad 100755
--- a/convlab/evaluator/multiwoz_eval.py
+++ b/convlab/evaluator/multiwoz_eval.py
@@ -461,6 +461,7 @@ class MultiWozEvaluator(Evaluator):
inform_sess[1] == 1 or inform_sess[1] is None) else 0
self.success = 0
self.success_strict = 0
+ print("goal", self.goal)
return 0
def domain_reqt_inform_analyze(self, domain, ref2goal=True):
diff --git a/convlab/policy/genTUS/evaluate.py b/convlab/policy/genTUS/evaluate.py
index 823b516c..87de8549 100644
--- a/convlab/policy/genTUS/evaluate.py
+++ b/convlab/policy/genTUS/evaluate.py
@@ -61,7 +61,6 @@ class Evaluator:
"gen_acts": [],
"gen_utts": []
}
- i = 0
for dialog in tqdm(in_file['dialog']):
inputs = dialog["in"]
labels = self.usr._parse_output(dialog["out"])
@@ -80,9 +79,6 @@ class Evaluator:
r["golden_utts"].append(labels["text"])
r["gen_acts"].append(usr_act)
r["gen_utts"].append(usr_utt)
- if i > 10:
- break
- i += 1
return r
diff --git a/convlab/policy/genTUS/stepGenTUS.py b/convlab/policy/genTUS/stepGenTUS.py
index da9840bd..2dd1c0a9 100644
--- a/convlab/policy/genTUS/stepGenTUS.py
+++ b/convlab/policy/genTUS/stepGenTUS.py
@@ -346,6 +346,7 @@ class UserActionPolicy(Policy):
self._read_goal(goal)
self.vector.init_session(goal=self.goal)
+ print("goal", self.goal)
self.terminated = False
self.add_sys_from_reward = False
--
GitLab