SVDTransformer

class lightautoml.transformers.decomposition.SVDTransformer(subs=None, random_state=42, n_components=100)[source]

Bases: lightautoml.transformers.base.LAMLTransformer

TruncatedSVD.

property features

Features list.

Return type

List[str]

__init__(subs=None, random_state=42, n_components=100)[source]
Parameters

  • subs (Optional[int]) – Subsample to fit algorithm. If None - full data.

  • random_state (int) – Random state to take subsample.

  • n_components (int) – Number of SVD components.

fit(dataset)[source]

Fit algorithm on dataset.

Parameters

dataset (NumpyDataset) – Sparse or Numpy dataset of text features.

transform(dataset)[source]

Transform input dataset to SVD representation.

Parameters

dataset (NumpyDataset) – Sparse or Numpy dataset of text features.

Return type

NumpyDataset

Returns

Numpy dataset with text embeddings.