diff --git a/examples/1_feature_overview/example.py b/examples/1_feature_overview/example.py index 411b619a0..71b9b0d1b 100644 --- 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)