Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PowerTransformer, QuantileTransformer and KernelCenterer #4755

Merged
6 changes: 6 additions & 0 deletions docs/source/api.rst
Original file line number Diff line number Diff line change
Expand Up @@ -108,10 +108,16 @@ Other preprocessing methods (Single-GPU)
:members:
.. autoclass:: cuml.preprocessing.KBinsDiscretizer
:members:
.. autoclass:: cuml.preprocessing.KernelCenterer
:members:
.. autoclass:: cuml.preprocessing.MissingIndicator
:members:
.. autoclass:: cuml.preprocessing.PolynomialFeatures
:members:
.. autoclass:: cuml.preprocessing.PowerTransformer
:members:
.. autoclass:: cuml.preprocessing.QuantileTransformer
:members:
.. autoclass:: cuml.preprocessing.SimpleImputer
:members:
.. autofunction:: cuml.preprocessing.add_dummy_feature
Expand Down
9 changes: 7 additions & 2 deletions python/cuml/_thirdparty/sklearn/preprocessing/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,13 @@


from ._data import Binarizer
from ._data import KernelCenterer
from ._data import MinMaxScaler
from ._data import MaxAbsScaler
from ._data import Normalizer
from ._data import PolynomialFeatures
from ._data import PowerTransformer
from ._data import QuantileTransformer
from ._data import RobustScaler
from ._data import StandardScaler
from ._data import add_dummy_feature
Expand All @@ -16,7 +20,8 @@
from ._data import robust_scale
from ._data import maxabs_scale
from ._data import minmax_scale
from ._data import PolynomialFeatures
from ._data import power_transform
from ._data import quantile_transform

from ._imputation import SimpleImputer
from ._imputation import MissingIndicator
Expand Down Expand Up @@ -57,8 +62,8 @@
'maxabs_scale',
'minmax_scale',
'label_binarize',
'quantile_transform',
'power_transform',
'quantile_transform',
'make_column_selector',
'make_column_transformer'
]