modlee.recommender.image_recommender module

class modlee.recommender.image_recommender.ImageClassificationRecommender(num_classes=None, *args, **kwargs)

Bases: ImageRecommender

Recommender for image classification tasks. Uses cross-entropy loss.

class modlee.recommender.image_recommender.ImageRecommender(*args, **kwargs)

Bases: Recommender

Recommender for image models.

calculate_metafeatures(dataloader, *args, **kwargs)

Calculate metafeatures.

Parameters:

dataloader – The dataloader on which to calculate metafeatures.

Returns:

The metafeatures of the data as a dictionary.

fit(dataloader, *args, **kwargs)

Fit the recommended to an image dataloader.

Parameters:

dataloader – The dataloader, should contain images as the first batch element.