NaNFlags

class lightautoml.transformers.numeric.NaNFlags(nan_rate=0.005)

Bases: lightautoml.transformers.base.LAMLTransformer

Create NaN flags.

__init__(nan_rate=0.005)

Parameters

nan_rate (float) – Nan rate cutoff.

fit(dataset)

Extract nan flags.

Parameters

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

Returns

self.

transform(dataset)

Transform - extract null flags.

Parameters

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

Return type

NumpyDataset

Returns

Numpy dataset with encoded labels.