Shortcuts

ProbabilitiesOutput

class flash.core.classification.ProbabilitiesOutput(multi_label=False)[source]

A Output which applies a softmax to the model outputs (assumed to be logits) and converts to a list.