()
| 11 | return data_dict |
| 12 | |
| 13 | def _parse_args(): |
| 14 | parser = argparse.ArgumentParser() |
| 15 | parser.add_argument('--dataset_type', default="WebQSP", type=str, help="CWQ | WebQSP") |
| 16 | args = parser.parse_args() |
| 17 | return args |
| 18 | |
| 19 | def prepare_dataloader(args,split): |
| 20 | assert split in ['train','test','dev','train_sample','dev_sample','test_sample'] |