Shortcuts

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.

Read the Docs v: stable
Versions
latest
stable
0.8.1
0.8.0
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.2
0.5.1
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.