TargetEncoder
- class lightautoml.transformers.categorical.TargetEncoder(alphas=(0.5, 1.0, 2.0, 5.0, 10.0, 50.0, 250.0, 1000.0))[source]
Bases:
LAMLTransformer
Out-of-fold target encoding.
Limitation:
Required .folds attribute in dataset - array of int from 0 to n_folds-1.
Working only after label encoding.
- fit_transform(dataset)[source]
Calc oof encoding and save encoding stats for new data.
- Parameters:
dataset (
Union
[NumpyDataset
,PandasDataset
]) – Pandas or Numpy dataset of categorical label encoded features.- Return type:
- Returns:
NumpyDataset - target encoded features.
- transform(dataset)[source]
Transform categorical dataset to target encoding.
- Parameters:
dataset (
Union
[NumpyDataset
,PandasDataset
]) – Pandas or Numpy dataset of categorical features.- Return type:
- Returns:
Numpy dataset with encoded labels.