Unverified Commit 0a64d942 authored by xuehui's avatar xuehui Committed by GitHub
Browse files

update typo in FeatureEngineering (#2292)

parent f36b62a9
......@@ -148,7 +148,7 @@ from sklearn.feature_selection.base import SelectorMixin
from nni.feature_engineering.feature_selector import FeatureSelector
class CustomizedSelector(FeatureSelector, BaseEstimator):
class CustomizedSelector(FeatureSelector, BaseEstimator, SelectorMixin):
def __init__(self, ...):
...
......@@ -161,7 +161,7 @@ class CustomizedSelector(FeatureSelector, BaseEstimator):
if not key.endswith('_')}
return params
def set_params(self, **params):
def set_params(self, **params):
"""
Set the parameters of this estimator.
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment