TextBertFeatures

class lightautoml.pipelines.features.text_pipeline.TextBertFeatures(**kwargs)[source]

Bases: FeaturesPipeline, NLPDataFeatures

Features pipeline for BERT.

create_pipeline(train)[source]

Create pipeline for BERT.

Parameters:

train (LAMLDataset) – Dataset with train data.

Return type:

LAMLTransformer

Returns:

Transformer.