Skip to content

Classification

Bases: TargetModel

Source code in autodistill/classification/classification_target_model.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
class ClassificationTargetModel(TargetModel):
    @abstractmethod
    def __init__(self):
        pass

    @abstractmethod
    def predict(self, input: str, confidence: float = 0.5) -> sv.Classifications:
        pass

    @abstractmethod
    def train(self):
        pass