LabelsOutput¶ class flash.core.classification.LabelsOutput(labels=None, multi_label=False, threshold=0.5)[source]¶ A Output which converts the model outputs (either logits or probabilities) to the label of the argmax classification. Parameters labels¶ (Optional[List[str]]) – A list of labels, assumed to map the class index to the label for that class. multi_label¶ (bool) – If true, treats outputs as multi label logits. threshold¶ (float) – The threshold to use for multi_label classification.