| 488 | |
| 489 | |
| 490 | def parse_args(): |
| 491 | parser = argparse.ArgumentParser() |
| 492 | |
| 493 | parser.add_argument("--dataset", type=str, default="All_Beauty", |
| 494 | help="Dataset name (folder name)") |
| 495 | |
| 496 | parser.add_argument("--user_k", type=int, default=5, |
| 497 | help="K-core threshold for users and items") |
| 498 | |
| 499 | parser.add_argument("--st_year", type=int, default=1996) |
| 500 | parser.add_argument("--st_month", type=int, default=1) |
| 501 | parser.add_argument("--ed_year", type=int, default=2025) |
| 502 | parser.add_argument("--ed_month", type=int, default=1) |
| 503 | |
| 504 | parser.add_argument("--metadata_file", type=str, required=True, |
| 505 | help="Path to meta_...jsonl file") |
| 506 | |
| 507 | parser.add_argument("--reviews_file", type=str, required=True, |
| 508 | help="Path to ...jsonl review file") |
| 509 | |
| 510 | parser.add_argument("--output_path", type=str, default="./data", |
| 511 | help="Output folder") |
| 512 | |
| 513 | return parser.parse_args() |
| 514 | |
| 515 | |
| 516 | def main(): |