File size: 1,224 Bytes
adc02fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | #!/usr/bin/env python
from __future__ import annotations
import argparse
import sys
from pathlib import Path
PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
sys.path.insert(0, str(PROJECT_ROOT))
from dovla_cil.experiments.reports import generate_dataset_report # noqa: E402
def main(argv: list[str] | None = None) -> int:
parser = argparse.ArgumentParser(description="Create analysis reports for a CIL dataset.")
parser.add_argument("--dataset", type=Path, required=True, help="CIL dataset directory.")
parser.add_argument("--out", type=Path, required=True, help="Report output directory.")
parser.add_argument("--sample-groups", type=int, default=3)
parser.add_argument("--seed", type=int, default=0)
args = parser.parse_args(argv)
summary = generate_dataset_report(
args.dataset,
args.out,
sample_groups=args.sample_groups,
seed=args.seed,
)
print(f"report: {args.out}")
print(f"num records: {summary['num_records']}")
print(f"num groups: {summary['num_groups']}")
print(f"success rate: {summary['success_rate']:.3f}")
return 0
if __name__ == "__main__":
raise SystemExit(main())
|