| |
| from __future__ import annotations |
|
|
| import argparse |
| import json |
| from pathlib import Path |
|
|
| from literature.evaluation import evaluate_predictions, load_json_records |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser(description="Evaluate extraction output against a POLYIE-style gold file.") |
| parser.add_argument("--gold", required=True, help="Gold file (.json or .jsonl)") |
| parser.add_argument("--pred", required=True, help="Prediction file (.json or .jsonl)") |
| parser.add_argument("--out", default=None, help="Optional JSON output path") |
| args = parser.parse_args() |
|
|
| gold_records = load_json_records(args.gold) |
| predicted_records = load_json_records(args.pred) |
| metrics = evaluate_predictions(gold_records, predicted_records) |
| text = json.dumps(metrics, indent=2, ensure_ascii=False) |
| print(text) |
|
|
| if args.out: |
| Path(args.out).write_text(text + "\n", encoding="utf-8") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
