BUG: fix example for using thresholds in SR3

dynamicslab · Aug 30, 2024 · 870525d · 870525d
1 parent 5ee3bcc
commit 870525d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/1_feature_overview/example.py b/examples/1_feature_overview/example.py
@@ -406,7 +406,7 @@ def f(x):
 model.print()
 
 # With thresholds matrix
-thresholds = 2 * np.ones((10, 3))
+thresholds = 2 * np.ones((3, 10))
 thresholds[4:, :] = 0.1
 sr3_optimizer = ps.SR3(thresholder="weighted_l0", thresholds=thresholds)
 model = ps.SINDy(optimizer=sr3_optimizer).fit(x_train, t=dt)