Skip to content
Snippets Groups Projects
Unverified Commit cce4fd8b authored by zhangzthu's avatar zhangzthu Committed by GitHub
Browse files

Fix a bug in TRADE CrossWOZ training (#138)


* add 'book' in DST evaluation.

* Fix TRADE crosswoz training evaluation bug

Co-authored-by: default avatarzheng <zheng@zhangzheng-PC.lan>
parent f5dd1ba3
Branches
No related tags found
No related merge requests found
......@@ -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]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment