From 870525daee52f596721caf7c5d5e479ad1fc9596 Mon Sep 17 00:00:00 2001 From: himkwtn Date: Fri, 30 Aug 2024 16:59:41 -0700 Subject: [PATCH] BUG: fix example for using thresholds in SR3 --- examples/1_feature_overview/example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)