Skip to content
Snippets Groups Projects
Commit 1211183d authored by function2's avatar function2
Browse files

Merge branch 'master' of github.com:thu-coai/ConvLab-2

parents 62a7d11b 1363ecdc
No related branches found
No related tags found
No related merge requests found
...@@ -45,6 +45,7 @@ class PolicyDataLoaderCrossWoz(): ...@@ -45,6 +45,7 @@ class PolicyDataLoaderCrossWoz():
dst.init_session() dst.init_session()
for i, turn in enumerate(sess): for i, turn in enumerate(sess):
if turn['role'] == 'usr': if turn['role'] == 'usr':
dst.state['user_action'] = turn['dialog_act']
dst.update(usr_da=turn['dialog_act']) dst.update(usr_da=turn['dialog_act'])
if i + 2 == len(sess): if i + 2 == len(sess):
dst.state['terminated'] = True dst.state['terminated'] = True
......
...@@ -114,7 +114,7 @@ class RuleBasedMultiwozBot(Policy): ...@@ -114,7 +114,7 @@ class RuleBasedMultiwozBot(Policy):
# print("Sys action: ", DA) # print("Sys action: ", DA)
if DA == {}: if len([domain_intent for domain_intent, slots in DA.items() if slots or 'nooffer' in domain_intent.lower()]) == 0:
DA = {'general-greet': [['none', 'none']]} DA = {'general-greet': [['none', 'none']]}
tuples = [] tuples = []
for domain_intent, svs in DA.items(): for domain_intent, svs in DA.items():
...@@ -201,6 +201,17 @@ class RuleBasedMultiwozBot(Policy): ...@@ -201,6 +201,17 @@ class RuleBasedMultiwozBot(Policy):
self.choice = "" self.choice = ""
elif self.recommend_flag == 1: elif self.recommend_flag == 1:
self.recommend_flag == 0 self.recommend_flag == 0
if len(kb_result) == 0:
if (domain + "-NoOffer") not in DA:
DA[domain + "-NoOffer"] = []
for slot in state['belief_state'][domain.lower()]['semi']:
if state['belief_state'][domain.lower()]['semi'][slot] != "" and \
state['belief_state'][domain.lower()]['semi'][slot] not in ["do nt care", "do n't care",
"dontcare"]:
slot_name = REF_USR_DA[domain].get(slot, slot)
DA[domain + "-NoOffer"].append([slot_name, state['belief_state'][domain.lower()]['semi'][slot]])
if (domain + "-Inform") not in DA: if (domain + "-Inform") not in DA:
DA[domain + "-Inform"] = [] DA[domain + "-Inform"] = []
for slot in user_action[user_act]: for slot in user_action[user_act]:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment