| 792 | |
| 793 | |
| 794 | def parse_args() -> argparse.Namespace: |
| 795 | parser = argparse.ArgumentParser(description="Run the Discourse-Aware main-experiment baseline.") |
| 796 | parser.add_argument("--input-dir", required=True, help="Clean baseline input directory.") |
| 797 | parser.add_argument( |
| 798 | "--output-dir", |
| 799 | default=None, |
| 800 | help="Output directory. Defaults to <input-dir>/../main_experiment/discourse_aware.", |
| 801 | ) |
| 802 | parser.add_argument( |
| 803 | "--roles-file", |
| 804 | default=str(PROJECT_ROOT / "data" / "roles.json"), |
| 805 | help="Initial role/profile metadata used for cold-start construction.", |
| 806 | ) |
| 807 | parser.add_argument("--top-k", type=int, default=DEFAULT_TOP_K, help="Number of papers shown per episode.") |
| 808 | return parser.parse_args() |
| 809 | |
| 810 | |
| 811 | def main() -> None: |