ClassesOutput¶ class flash.core.classification.ClassesOutput(multi_label=False, threshold=0.5)[source]¶ A Output which applies an argmax to the model outputs (either logits or probabilities) and converts to a list. Parameters multi_label¶ (bool) – If true, treats outputs as multi label logits. threshold¶ (float) – The threshold to use for multi_label classification.