LabelEncoder
- class lightautoml.transformers.categorical.LabelEncoder(subs=None, random_state=42)[source]
Bases:
lightautoml.transformers.base.LAMLTransformer
Simple LabelEncoder in order of frequency.
Labels are integers from 1 to n. Unknown category encoded as 0. NaN is handled as a category value.
- fit(dataset)[source]
Estimate label frequencies and create encoding dicts.
- Parameters
dataset (
Union
[NumpyDataset
,PandasDataset
]) – Pandas or Numpy dataset of categorical features.- Returns
self.
- transform(dataset)[source]
Transform categorical dataset to int labels.
- Parameters
dataset (
Union
[NumpyDataset
,PandasDataset
]) – Pandas or Numpy dataset of categorical features.- Return type
- Returns
Numpy dataset with encoded labels.