Add _ridge function for ridge regression

Parietal-INRIA · Jan 19, 2024 · 9b38655 · 9b38655
1 parent f85b1c6
commit 9b38655
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/fmralign/hyperalignment/linalg.py b/fmralign/hyperalignment/linalg.py
@@ -108,6 +108,14 @@ def ridge(X, Y, alpha):
     return betas
 
 
+def _ridge(X, Y, alpha):
+    from sklearn.linear_model import Ridge
+
+    ridge = Ridge(alpha=alpha, fit_intercept=False)
+    ridge.fit(X, Y)
+    return ridge.coef_.T
+
+
 def procrustes(X, Y, reflection=True, scaling=False):
     r"""
     The orthogonal Procrustes algorithm.