From a067adb84a0175f28f41cc192b6a6e31a497e654 Mon Sep 17 00:00:00 2001
From: zhuqi <zqwerty@users.noreply.github.com>
Date: Fri, 7 May 2021 20:07:07 +0800
Subject: [PATCH] Update goal_generator.py (#194)
add back postcode for policy, add postcode and address for hospital
#191
---
convlab2/task/multiwoz/goal_generator.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/convlab2/task/multiwoz/goal_generator.py b/convlab2/task/multiwoz/goal_generator.py
index 49c10f6..6a9073b 100755
--- a/convlab2/task/multiwoz/goal_generator.py
+++ b/convlab2/task/multiwoz/goal_generator.py
@@ -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'])
--
GitLab