fix: return intercept as float

jolars · Dec 11, 2023 · 5e83de1 · 5e83de1
1 parent afe26a5
commit 5e83de1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sortedl1/estimators.py b/sortedl1/estimators.py
@@ -104,7 +104,7 @@ def fit(self, X, y):
 
         result = fit_slope(X, y, lam, alpha, params)
 
-        self.intercept_ = result[0]
+        self.intercept_ = result[0][0]
         self.sparse_coef_ = result[1]
         self.coef_ = result[1].toarray()
         self.lambda_ = result[2]