Skip to content
Snippets Groups Projects
Unverified Commit 1363ecdc authored by QD0716's avatar QD0716 Committed by GitHub
Browse files

Fix bug for can't generate correct vectorized state (#152)

dst.update don't update dst.state['user_action'], so dst.state['user_action'] is always empty, The following function state_vectorize cannot get the correct user_action. sr_act_vec is always zero.
parent db2d0971
No related branches found
No related tags found
No related merge requests found
......@@ -45,6 +45,7 @@ class PolicyDataLoaderCrossWoz():
dst.init_session()
for i, turn in enumerate(sess):
if turn['role'] == 'usr':
dst.state['user_action'] = turn['dialog_act']
dst.update(usr_da=turn['dialog_act'])
if i + 2 == len(sess):
dst.state['terminated'] = True
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment