From 927241c496d2508344f463c9cf96b280b5cc2830 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Mon, 10 Aug 2026 04:31:50 -0500 Subject: feat: add local online least-squares residual predictor --- experiments/rain_ep_bias_train.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'experiments/rain_ep_bias_train.py') diff --git a/experiments/rain_ep_bias_train.py b/experiments/rain_ep_bias_train.py index d6f1f85..70ff441 100644 --- a/experiments/rain_ep_bias_train.py +++ b/experiments/rain_ep_bias_train.py @@ -60,6 +60,8 @@ def parse_args() -> argparse.Namespace: "--dillavou-profile-json", type=Path, help="released physical state-dependence report; omitted means constant B") parser.add_argument("--predictor-rate", type=float, default=0.1) + parser.add_argument( + "--predictor-kind", choices=("nlms", "ols"), default="nlms") parser.add_argument( "--neutral-cadence", type=int, default=1, help="training steps per neutral update; zero freezes after calibration") @@ -247,6 +249,7 @@ def main() -> None: neutral_cadence=args.neutral_cadence, drift_ratio=args.dillavou_drift_ratio, empirical_profile=empirical_profile, + predictor_kind=args.predictor_kind, seed=args.seed + 1729, ) attach_dillavou_to_rain_estimator(estimator, corrector) @@ -370,6 +373,7 @@ def main() -> None: or corrector.empirical_profile is None else corrector.empirical_profile.as_dict()), "predictor_rate": args.predictor_rate, + "predictor_kind": args.predictor_kind, "neutral_cadence": args.neutral_cadence, "layer_calibration_steps": args.layer_calibration_steps, "layer_bias_normalization": args.layer_bias_normalization, -- cgit v1.2.3