Skip to content

Commit a130b8a

Browse files
committed
Merge pull request scikit-learn#3349 from MechCoder/return_niter
[MRG+1] Return attribute n_iter_ for linear models dependent on the enet solver
2 parents 228bc99 + d8ecd15 commit a130b8a

File tree

5 files changed

+1888
-1900
lines changed

5 files changed

+1888
-1900
lines changed

sklearn/covariance/graph_lasso_.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def graph_lasso(emp_cov, alpha, cov_init=None, mode='cd', tol=1e-4,
187187
# Use coordinate descent
188188
coefs = -(precision_[indices != idx, idx]
189189
/ (precision_[idx, idx] + 1000 * eps))
190-
coefs, _, _ = cd_fast.enet_coordinate_descent_gram(
190+
coefs, _, _, _ = cd_fast.enet_coordinate_descent_gram(
191191
coefs, alpha, 0, sub_covariance, row, row,
192192
max_iter, tol)
193193
else:

0 commit comments

Comments
 (0)