()
| 5 | from argparse import ArgumentParser |
| 6 | |
| 7 | def get_args(): |
| 8 | parser = ArgumentParser(description='Data preprocess') |
| 9 | parser.add_argument('--path', type=str, default='data', help='Path to the SocraTeach dataset') |
| 10 | parser.add_argument('--split_fold', type=str, default='data/data_split', help='Folder path to save the split train/valid/test subsets') |
| 11 | args = parser.parse_args() |
| 12 | return args |
| 13 | args = get_args() |
| 14 | os.makedirs(args.split_fold, exist_ok=True) |
| 15 |