From 1f8ed9fd76c74359c2fc9cfec790b5eed4b66fba Mon Sep 17 00:00:00 2001
From: zhangzthu <zhangz.goal@gmail.com>
Date: Sun, 20 Sep 2020 15:35:23 +0800
Subject: [PATCH] add 'book' in DST evaluation. (#85)
---
convlab2/dst/evaluate.py | 17 ++++++++++++++---
1 file changed, 14 insertions(+), 3 deletions(-)
diff --git a/convlab2/dst/evaluate.py b/convlab2/dst/evaluate.py
index 842dde7..04263a1 100755
--- a/convlab2/dst/evaluate.py
+++ b/convlab2/dst/evaluate.py
@@ -56,13 +56,24 @@ def reformat_state(state):
state = state['belief_state']
new_state = []
for domain in state.keys():
- domain_data = state[domain]
- if 'semi' in domain_data:
- domain_data = domain_data['semi']
+ domain_data_all = state[domain]
+ if 'semi' in domain_data_all:
+ domain_data = domain_data_all['semi']
for slot in domain_data.keys():
val = domain_data[slot]
if val is not None and val not in ['', 'not mentioned', '未提及', '未提到', '没有提到']:
new_state.append(domain + '-' + slot + '-' + val)
+ if 'book' in domain_data_all:
+ domain_data = domain_data_all['book']
+ for slot in domain_data.keys():
+ if slot == 'booked':
+ continue
+ elif domain == 'bus' and slot == 'people':
+ continue
+ else:
+ val = domain_data[slot]
+ if val is not None and val not in ['', 'not mentioned', '未提及', '未提到', '没有提到']:
+ new_state.append(domain+'_book' + '-' + slot + '-' + val)
# lower
new_state = [item.lower() for item in new_state]
return new_state
--
GitLab