use eigen leads for solving lasso

braindatalab · Dec 11, 2024 · 36e22b9 · 36e22b9
1 parent 515539e
commit 36e22b9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bsi_zoo/estimators.py b/bsi_zoo/estimators.py
@@ -594,10 +594,12 @@ def gprime(w):
 
     alpha = alpha * alpha_max
 
+    eigen_fields, sing, eigen_leads = _safe_svd(L, full_matrices=False)
+
     # y->M
     # L->gain
     x = _solve_reweighted_lasso(
-        L, y, alpha, n_orient, weights, max_iter, max_iter_reweighting, gprime
+        eigen_leads, y, alpha, n_orient, weights, max_iter, max_iter_reweighting, gprime
     )
 
     return x