BUG: fix l0 reg

dynamicslab · Aug 20, 2024 · 4509ea2 · 4509ea2
1 parent 3cad8f6
commit 4509ea2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysindy/utils/base.py b/pysindy/utils/base.py
@@ -232,7 +232,7 @@ def get_prox(
 
 
 def regularization_l0(x: NDArray[np.float64], lam: NDArray[np.float64]):
-    return lam.item() * np.count_nonzero(x)
+    return lam * np.count_nonzero(x)
 
 
 def regualization_weighted_l0(x: NDArray[np.float64], lam: NDArray[np.float64]):