Update evaluate_checkpoint.py

evaluate_checkpoint.py

@@ -56,7 +56,6 @@ def _episode_is_risky(
     clean_ratio: float,
     spatial_corr: float,
 ) -> bool:
-    """Match training: regional event plan, not independent per-zone draws."""
     plan = _episode_event_plan(
         n_zones, effective_seed, clean_ratio, spatial_corr
     )
@@ -474,6 +473,24 @@ def main() -> None:
         default=0.85,
         help="Must match train",
     )
+    p.add_argument(
+        "--alert-value",
+        type=float,
+        default=100.0,
+        help="Must match train (default 100.0)",
+    )
+    p.add_argument(
+        "--false-alert-penalty",
+        type=float,
+        default=20.0,
+        help="Must match train (default 20.0)",
+    )
+    p.add_argument(
+        "--miss-penalty",
+        type=float,
+        default=50.0,
+        help="Must match train (default 50.0)",
+    )
     args = p.parse_args()
 
     if args.policy == "checkpoint" and not args.checkpoint:
@@ -485,6 +502,9 @@ def main() -> None:
         soft_reset=True,
         clean_episode_ratio=args.clean_episode_ratio,
         event_spatial_correlation=args.event_spatial_correlation,
+        alert_value=args.alert_value,
+        false_alert_penalty=args.false_alert_penalty,
+        miss_penalty=args.miss_penalty,
     )
 
     model = None