modlee.recommender package

Submodules

• modlee.recommender.image_recommender module
  • ImageClassificationRecommender
  • ImageRecommender
    • ImageRecommender.fit()
  • ImageSegmentationRecommender
• modlee.recommender.recommender module
  • Recommender
    • Recommender.analyze()
    • Recommender.calculate_metafeatures()
    • Recommender.fit()
    • Recommender.get_code_text()
    • Recommender.get_input_torch()
    • Recommender.get_model_details()
    • Recommender.model
    • Recommender.train()
    • Recommender.train_documentation_locations()
    • Recommender.train_documentation_shared()
    • Recommender.write_file()

Module contents

Recommender for models.

modlee.recommender.from_modality_task(modality, task, *args, **kwargs)

Return a Recommender object based on the modality and task. Currently supports:

• image

— classification — segmentation - text — classification

Parameters:

• modality – The modality as a string, e.g. “image”, “text”.

• task – The task as a string, e.g. “classification”, “segmentation”.

Returns:

The RecommenderObject, if it exists.