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

Update goal_generator.py (#194) 

add back postcode for policy, add postcode and address for hospital
#191
parent 1baf20ff
Branches
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 @ a067adb8
...@@ -166,13 +166,13 @@ class GoalGenerator: ...@@ -166,13 +166,13 @@ class GoalGenerator:
self._build_goal_model() self._build_goal_model()
print('Building goal model is done') print('Building goal model is done')
# remove some slot # 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_dist['police']['reqt']['postcode']
del self.ind_slot_value_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_dist['hospital']['reqt']['postcode']
del self.ind_slot_value_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_dist['hospital']['reqt']['address']
del self.ind_slot_value_dist['hospital']['reqt']['address'] # del self.ind_slot_value_dist['hospital']['reqt']['address']
# print(self.slots_combination_dist['police']) # print(self.slots_combination_dist['police'])
# print(self.slots_combination_dist['hospital']) # 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