ImageAutoFeatures

class lightautoml.pipelines.features.image_pipeline.ImageAutoFeatures(**kwargs)

Bases: FeaturesPipeline, ImageDataFeatures

Class contains efficient-net embeddings features for image data.

create_pipeline(train)

Create pipeline for images data.

Parameters:

train (LAMLDataset) – Dataset with train features.

Return type:

LAMLTransformer

Returns:

Transformer.