Skip to content

Commit ae74c9d

Browse files
veneamueller
authored andcommitted
Use same parameters when solving the transform
1 parent 206d227 commit ae74c9d

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

sklearn/decomposition/nmf.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
import numbers
1616

1717
import numpy as np
18-
from scipy.optimize import nnls
1918
import scipy.sparse as sp
2019

2120
from ..base import BaseEstimator, TransformerMixin
@@ -572,10 +571,8 @@ def transform(self, X):
572571
"""
573572
X, = check_arrays(X, sparse_format='csc')
574573
Wt = np.zeros((self.n_components_, X.shape[0]))
575-
Wt, _, _ = _nls_subproblem(X.T, self.components_.T, Wt, tol=1e-10,
576-
max_iter=1000000)
577-
#for j in range(0, X.shape[0]):
578-
# W[j, :], _ = nnls(self.components_.T, X[j, :])
574+
Wt, _, _ = _nls_subproblem(X.T, self.components_.T, Wt, tol=self.tol,
575+
max_iter=self.nls_max_iter)
579576
return Wt.T
580577

581578

0 commit comments

Comments
 (0)