Skip to content
Snippets Groups Projects
Unverified Commit 641eb4cd authored by zhuqi's avatar zhuqi Committed by GitHub
Browse files

Update goal_generator.py 

add back postcode for policy, add postcode and address for hospital
#191
parent 1baf20ff
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
convlab2/task/multiwoz/goal_generator.py
+ 7
− 7
View file @ 641eb4cd
......@@ -166,13 +166,13 @@ class GoalGenerator:
self._build_goal_model()
print('Building goal model is done')
# remove some slot
del self.ind_slot_dist['police']['reqt']['postcode']
del self.ind_slot_value_dist['police']['reqt']['postcode']
del self.ind_slot_dist['hospital']['reqt']['postcode']
del self.ind_slot_value_dist['hospital']['reqt']['postcode']
del self.ind_slot_dist['hospital']['reqt']['address']
del self.ind_slot_value_dist['hospital']['reqt']['address']
# remove some slot. Fixed: add postcode for policy db, add postcode and address for hospital db
# del self.ind_slot_dist['police']['reqt']['postcode']
# del self.ind_slot_value_dist['police']['reqt']['postcode']
# del self.ind_slot_dist['hospital']['reqt']['postcode']
# del self.ind_slot_value_dist['hospital']['reqt']['postcode']
# del self.ind_slot_dist['hospital']['reqt']['address']
# del self.ind_slot_value_dist['hospital']['reqt']['address']
# print(self.slots_combination_dist['police'])
# print(self.slots_combination_dist['hospital'])
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment