def get_train_examples(self, data_dir, aug_file, args):
if (aug_file):
return dataset_multiwoz21.create_aug_examples(
os.path.join(data_dir, 'train_dials.json'),
os.path.join(data_dir, 'dialogue_acts.json'), 'train',
self.slot_list, self.label_maps, aug_file, **args)
else:
return dataset_multiwoz21.create_examples(
os.path.join(data_dir, 'train_dials.json'),
os.path.join(data_dir, 'dialogue_acts.json'), 'train',
self.slot_list, self.label_maps, **args)
def get_test_examples(self, data_dir, args):
return dataset_multiwoz21.create_examples(
os.path.join(data_dir, 'test_dials.json'),
os.path.join(data_dir, 'dialogue_acts.json'), 'test',
self.slot_list, self.label_maps, **args)