missing argument

Added missing deep argument to get_params of LowRankLogisticRegression
This commit is contained in:
Andrea Esuli 2023-02-28 08:41:34 +01:00 committed by GitHub
parent 140ab3bfc9
commit e9d56e5801
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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):