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)