()
| 465 | |
| 466 | |
| 467 | def parse_args(): |
| 468 | parser = argparse.ArgumentParser() |
| 469 | parser.add_argument('--dataset', type=str, default='Arts', help='Instruments / Arts / Games / Sports') |
| 470 | parser.add_argument('--user_k', type=int, default=5, help='user k-core filtering') |
| 471 | parser.add_argument('--item_k', type=int, default=5, help='item k-core filtering') |
| 472 | parser.add_argument('--st_year', type=int, default=1996, help='start year') |
| 473 | parser.add_argument('--st_month', type=int, default=10, help='start month') |
| 474 | parser.add_argument('--ed_year', type=int, default=2018, help='end year') |
| 475 | parser.add_argument('--ed_month', type=int, default=11, help='end month') |
| 476 | parser.add_argument('--metadata_file', type=str, default=None, help='metadata file path') |
| 477 | parser.add_argument('--reviews_file', type=str, default=None, help='reviews file path') |
| 478 | parser.add_argument('--output_path', type=str, default='./data', help='output directory') |
| 479 | return parser.parse_args() |
| 480 | |
| 481 | |
| 482 | if __name__ == '__main__': |
no outgoing calls
no test coverage detected