Add CI and shadow calibration reporting

2026-03-20 20:59:30 +08:00
parent 25ab512371
commit 1c84893bed
39 changed files with 149223 additions and 3277 deletions
@@ -0,0 +1,65 @@
+import argparse
+import json
+import os
+import sys
+
+PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+if PROJECT_ROOT not in sys.path:
+    sys.path.insert(0, PROJECT_ROOT)
+
+from src.analysis.deb_algorithm import load_history  # noqa: E402
+from scripts.fit_probability_calibration import (  # noqa: E402
+    _extract_samples,
+    _load_json_if_exists,
+)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Export normalized probability calibration training samples.")
+    parser.add_argument(
+        "--history-file",
+        default=os.path.join(PROJECT_ROOT, "data", "daily_records.json"),
+    )
+    parser.add_argument(
+        "--settlement-history",
+        default=os.path.join(
+            PROJECT_ROOT,
+            "artifacts",
+            "probability_calibration",
+            "settlement_history.json",
+        ),
+    )
+    parser.add_argument(
+        "--output",
+        default=os.path.join(
+            PROJECT_ROOT,
+            "artifacts",
+            "probability_calibration",
+            "training_samples.json",
+        ),
+    )
+    args = parser.parse_args()
+
+    history = load_history(args.history_file)
+    settlement_history = _load_json_if_exists(args.settlement_history)
+    samples, filled_actual_from_history = _extract_samples(
+        history,
+        settlement_history=settlement_history,
+    )
+
+    output_dir = os.path.dirname(os.path.abspath(args.output))
+    if output_dir:
+        os.makedirs(output_dir, exist_ok=True)
+    payload = {
+        "sample_count": len(samples),
+        "filled_actual_from_history": filled_actual_from_history,
+        "samples": samples,
+    }
+    with open(args.output, "w", encoding="utf-8") as fh:
+        json.dump(payload, fh, ensure_ascii=False, indent=2)
+
+    print(f"exported {len(samples)} samples to {args.output}")
+
+
+if __name__ == "__main__":
+    main()