diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-10 04:31:50 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-08-10 04:31:50 -0500 |
| commit | 927241c496d2508344f463c9cf96b280b5cc2830 (patch) | |
| tree | f2a8bb017f305d57581103c71345a47ba09ca55f /experiments/rain_ep_bias_train.py | |
| parent | 29808ed4f8d550eb0ddcbc300b33f2dc596721e3 (diff) | |
feat: add local online least-squares residual predictor
Diffstat (limited to 'experiments/rain_ep_bias_train.py')
| -rw-r--r-- | experiments/rain_ep_bias_train.py | 4 |
1 files changed, 4 insertions, 0 deletions
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 @@ -61,6 +61,8 @@ def parse_args() -> argparse.Namespace: 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") parser.add_argument("--calibration-batches", type=int, default=0) @@ -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, |