Skip to content

Commit 9eb19ae

Browse files
committed
shorten sqrtLasso.coef_ by 1 coef if an intercept is fitted
1 parent a579b0c commit 9eb19ae

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

skglm/experimental/sqrt_lasso.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ def fit(self, X, y):
137137
self.coef_ = self.path(X, y, alphas=[self.alpha])[1][0]
138138
if self.fit_intercept:
139139
self.intercept_ = self.coef_[-1]
140+
self.coef_ = self.coef_[:-1]
140141
else:
141142
self.intercept_ = 0.
142143
return self

0 commit comments

Comments
 (0)