-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathbuild_data_dstc2.py
60 lines (51 loc) · 1.93 KB
/
build_data_dstc2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import json
import os
import data
import data_utils
import xtrack2_config
def main(skip_dstc_import_step, builder_type):
import utils
utils.pdb_on_error()
ontology_path = os.path.join(xtrack2_config.data_directory,
'dstc2/scripts/config/ontology_dstc2.json')
with open(ontology_path) as f_in:
dstc_ontology = json.load(f_in)
ontology = dict(
food=dstc_ontology['informable']['food'],
pricerange=dstc_ontology['informable']['pricerange'],
area=dstc_ontology['informable']['area'],
name=dstc_ontology['informable']['name'],
method=dstc_ontology['method']
)
data_utils.prepare_experiment(
experiment_name='e2_tagged_%s' % builder_type,
data_directory=xtrack2_config.data_directory,
slots=['food', 'area', 'pricerange', 'name', 'method', 'req_food',
'req_area', 'req_pricerange', 'req_name', 'req_phone',
'req_addr', 'req_postcode', 'req_signature'],
slot_groups=dict(
food=['food'],
area=['area'],
pricerange=['pricerange'],
name=['name'],
method=['method'],
goals=['food', 'area', 'pricerange', 'name'],
requested=['req_food', 'req_area', 'req_pricerange', 'req_name',
'req_phone', 'req_addr', 'req_postcode', 'req_signature']
),
ontology=ontology,
builder_opts=dict(
tagged=True,
no_label_weight=True
),
skip_dstc_import_step=skip_dstc_import_step,
builder_type=builder_type
)
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--skip_dstc_import_step', action='store_true',
default=False)
parser.add_argument('--builder_type', default='xtrack')
args = parser.parse_args()
main(**vars(args))