OrdinalEncoder

class lightautoml.transformers.categorical.OrdinalEncoder(*args, **kwargs)

Bases: LabelEncoder

Encoding ordinal categories into numbers.

Number type categories passed as is, object type sorted in ascending lexicographical order.

fit(dataset)

Estimate label frequencies and create encoding dicts.

Parameters:

dataset (Union[NumpyDataset, PandasDataset]) – Pandas or Numpy dataset of categorical features.

Returns:

Self.