ColumnsSelector

class lightautoml.transformers.base.ColumnsSelector(keys)

Bases: lightautoml.transformers.base.LAMLTransformer

Select columns to pass to another transformers (or feature selection).

__init__(keys)

Parameters

keys (Sequence[str]) – Columns names.

fit(dataset)

Empty fit method - just set features.

Parameters

dataset (LAMLDataset) – Dataset to fit on.

Return type

ColumnsSelector

Returns

self.

transform(dataset)

Select given keys from dataset.

Parameters

dataset (LAMLTransformer) – Dataset to transform.

Return type

LAMLTransformer

Returns

Dataset with selected features.