lightautoml.pipelines.features

Pipelines for features generation.

Base Classes

FeaturesPipeline	Abstract class.
EmptyFeaturePipeline	Dummy feature pipeline - .fit_transform and transform do nothing.
TabularDataFeatures	Helper class contains basic features transformations for tabular data.

Feature Pipelines for Boosting Models

LGBSimpleFeatures	Creates simple pipeline for tree based models.
LGBAdvancedPipeline	Create advanced pipeline for trees based models.

Feature Pipelines for Linear Models

LinearFeatures

Creates pipeline for linear models and nnets.

Feature Pipelines for WhiteBox

WBFeatures

Simple WhiteBox pipeline.

Image Feature Pipelines

ImageDataFeatures	Class contains basic features transformations for image data.
ImageSimpleFeatures	Class contains simple color histogram features for image data.
ImageAutoFeatures	Class contains efficient-net embeddings features for image data.

Text Feature Pipelines

NLPDataFeatures	Class contains basic features transformations for text data.
TextAutoFeatures	Class contains embedding features for text data.
NLPTFiDFFeatures	Class contains tfidf features for text data.
TextBertFeatures	Features pipeline for BERT.