DeepImageEmbedder
- class lightautoml.image.image.DeepImageEmbedder(device=torch.device, n_jobs=4, random_state=42, is_advprop=True, model_name='efficientnet-b0', weights_path=None, batch_size=128, verbose=True)[source]
Bases:
sklearn.base.TransformerMixin
Transformer for image embeddings.
- __init__(device=torch.device, n_jobs=4, random_state=42, is_advprop=True, model_name='efficientnet-b0', weights_path=None, batch_size=128, verbose=True)[source]
Pytorch Dataset for
EffNetImageEmbedder
.