From cce4fd8b0c4b9258c4c1bfc2cdee661f32409157 Mon Sep 17 00:00:00 2001
From: zhangzthu <zhangz.goal@gmail.com>
Date: Wed, 30 Sep 2020 16:06:27 +0800
Subject: [PATCH] Fix a bug in TRADE CrossWOZ training (#138)
* add 'book' in DST evaluation.
* Fix TRADE crosswoz training evaluation bug
Co-authored-by: zheng <zheng@zhangzheng-PC.lan>
---
convlab2/dst/trade/crosswoz/models/TRADE.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/convlab2/dst/trade/crosswoz/models/TRADE.py b/convlab2/dst/trade/crosswoz/models/TRADE.py
index e70e877..d0b9423 100755
--- a/convlab2/dst/trade/crosswoz/models/TRADE.py
+++ b/convlab2/dst/trade/crosswoz/models/TRADE.py
@@ -235,7 +235,6 @@ class TRADE(nn.Module):
predict_belief_bsz_ptr.append(slot_temp[si] + "-" + str(st))
all_prediction[data_dev["ID"][bi]][data_dev["turn_id"][bi]]["pred_bs_ptr"] = predict_belief_bsz_ptr
- return predict_belief_bsz_ptr
if set(data_dev["turn_belief"][bi]) != set(predict_belief_bsz_ptr) and args["genSample"]:
print("True", set(data_dev["turn_belief"][bi]))
--
GitLab