Source code for shifthappens.models.mixins

"""Model mixins indicate the supported prediction types of a model."""


[docs]class LabelModelMixin: """Inherit from this class if your model returns predicted labels.""" pass
[docs]class ConfidenceModelMixin: """Inherit from this class if your model returns confidences.""" pass
[docs]class UncertaintyModelMixin: """Inherit from this class if your model returns uncertainties.""" pass
[docs]class OODScoreModelMixin: """Inherit from this class if your model returns ood scores.""" pass
[docs]class FeaturesModelMixin: """Inherit from this class if your model returns features.""" pass