experiment: add one-sided residual stability margin

1 files changed, 9 insertions, 0 deletions

diff --git a/experiments/conv_local_smoke.py b/experiments/conv_local_smoke.py
index 802051b..dbae759 100644
--- a/experiments/conv_local_smoke.py
+++ b/experiments/conv_local_smoke.py
@@ -997,6 +997,15 @@ def kp_mixed_traffic_checks():
     for slope, bias in zip(net.P_traffic, net.P_traffic_bias):
         slope.zero_()
         bias.zero_()
+    stable_fit = net.predictor_closed_form_fit(
+        forward["hiddens"], stability_margin=1e-3)
+    assert stable_fit["max_positive_residual_soma_slope"] < 1e-14
+    assert stable_fit["min_residual_soma_slope"] < -9e-4
+    assert stable_fit["max_applied_stability_margin"] >= 1e-3
+
+    for slope, bias in zip(net.P_traffic, net.P_traffic_bias):
+        slope.zero_()
+        bias.zero_()
     components = net.mixed_apical_components(
         instruction, forward["hiddens"], "matched")
     norm_errors = []