TextBertFeatures

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

Bases: lightautoml.pipelines.features.base.FeaturesPipeline, lightautoml.pipelines.features.text_pipeline.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.