()
| 910 | |
| 911 | |
| 912 | def main() -> None: |
| 913 | args = parse_args() |
| 914 | input_dir = Path(args.input_dir) |
| 915 | output_dir = ( |
| 916 | Path(args.output_dir) |
| 917 | if args.output_dir |
| 918 | else input_dir.parent / "main_experiment" / METHOD_KEY |
| 919 | ) |
| 920 | result = run_baseline( |
| 921 | input_dir=input_dir, |
| 922 | output_dir=output_dir, |
| 923 | roles_file=Path(args.roles_file), |
| 924 | top_k=args.top_k, |
| 925 | ) |
| 926 | summary = result["evaluation"]["summary"] |
| 927 | print(json.dumps(summary, ensure_ascii=False, indent=2)) |
| 928 | print(f"\nKnowledge-Entity outputs written to: {output_dir}") |
| 929 | |
| 930 | |
| 931 | if __name__ == "__main__": |
no test coverage detected