missing argument
Added missing deep argument to get_params of LowRankLogisticRegression
This commit is contained in:
parent
140ab3bfc9
commit
e9d56e5801
|
@ -21,14 +21,14 @@ class LowRankLogisticRegression(BaseEstimator):
|
|||
self.n_components = n_components
|
||||
self.learner = LogisticRegression(**kwargs)
|
||||
|
||||
def get_params(self):
|
||||
def get_params(self, deep=True):
|
||||
"""
|
||||
Get hyper-parameters for this estimator.
|
||||
|
||||
:return: a dictionary with parameter names mapped to their values
|
||||
"""
|
||||
params = {'n_components': self.n_components}
|
||||
params.update(self.learner.get_params())
|
||||
params.update(self.learner.get_params(deep))
|
||||
return params
|
||||
|
||||
def set_params(self, **params):
|
||||
|
|
Loading…
Reference in New Issue