diff options
Diffstat (limited to 'experiments/rain_ep_dillavou_smoke.py')
| -rw-r--r-- | experiments/rain_ep_dillavou_smoke.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/experiments/rain_ep_dillavou_smoke.py b/experiments/rain_ep_dillavou_smoke.py index d4a634a..9519210 100644 --- a/experiments/rain_ep_dillavou_smoke.py +++ b/experiments/rain_ep_dillavou_smoke.py @@ -70,9 +70,11 @@ def main() -> None: assert fixed_relative_error < 2e-6, fixed_relative_error constant = DillavouUpdateCorrector( - mode="constant", bias_ratio=0.2, predictor_rate=1.0, seed=41) + mode="constant", bias_ratio=0.2, predictor_rate=1.0, + calibration_steps=1, neutral_cadence=0, seed=41) innovation = DillavouUpdateCorrector( - mode="innovation", bias_ratio=0.2, predictor_rate=1.0, seed=41) + mode="innovation", bias_ratio=0.2, predictor_rate=1.0, + calibration_steps=1, neutral_cadence=0, seed=41) corrected_constant = constant.apply(clean_a, parameters_a) corrected_innovation = innovation.apply(clean_a, parameters_a) constant_error = max( @@ -85,6 +87,10 @@ def main() -> None: ) assert constant_error < 2e-7, constant_error assert innovation_error < 2e-7, innovation_error + constant.apply(clean_b, parameters_b) + innovation.apply(clean_b, parameters_b) + assert constant.debiaser.neutral_observations == 1 + assert innovation.debiaser.neutral_observations == 1 # Integration check: the corruption is attached after Rain's hand-written # local EP estimator and introduces no autograd graph. |