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:
- Returns:
Transformer.