ColumnsSelector

class lightautoml.transformers.base.ColumnsSelector(keys)[source]

Bases: lightautoml.transformers.base.LAMLTransformer

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

__init__(keys)[source]
Parameters

keys (Sequence[str]) – Columns names.

fit(dataset)[source]

Empty fit method - just set features.

Parameters

dataset (LAMLDataset) – Dataset to fit on.

Return type

ColumnsSelector

Returns

self.

transform(dataset)[source]

Select given keys from dataset.

Parameters

dataset (LAMLDataset) – Dataset to transform.

Return type

LAMLDataset

Returns

Dataset with selected features.