ClassificationOutput

class flash.core.classification.ClassificationOutput(multi_label=False)

A base class for classification outputs.

Parameters

multi_label (bool) – If true, treats outputs as multi label logits.