| 893 | |
| 894 | |
| 895 | def parse_args() -> argparse.Namespace: |
| 896 | parser = argparse.ArgumentParser(description="Run the Knowledge-Entity main-experiment baseline.") |
| 897 | parser.add_argument("--input-dir", required=True, help="Clean baseline input directory.") |
| 898 | parser.add_argument( |
| 899 | "--output-dir", |
| 900 | default=None, |
| 901 | help="Output directory. Defaults to <input-dir>/../main_experiment/knowledge_entity.", |
| 902 | ) |
| 903 | parser.add_argument( |
| 904 | "--roles-file", |
| 905 | default=str(PROJECT_ROOT / "data" / "roles.json"), |
| 906 | help="Initial role/profile metadata used for cold-start construction.", |
| 907 | ) |
| 908 | parser.add_argument("--top-k", type=int, default=DEFAULT_TOP_K, help="Number of papers shown per episode.") |
| 909 | return parser.parse_args() |
| 910 | |
| 911 | |
| 912 | def main() -> None: |