ColumnsSelector

class lightautoml.transformers.base.ColumnsSelector(keys)

Bases: LAMLTransformer

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

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 (LAMLDataset) – Dataset to transform.

Return type:

LAMLDataset

Returns:

Dataset with selected features.