def is_classifier()

in src/exporters/coreml/config.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def is_classifier(self) -> bool:
        """
        Determines whether this is a treated as a special classifier model by Core ML.
        """
        classifier_tasks = [
            "image-classification",
            "multiple-choice",
            "next-sentence-prediction",
            "text-classification"
        ]
        return self.task in classifier_tasks and self.outputs["logits"].do_softmax