Table of Contents

Shortcuts

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.

Read the Docs v: stable

Versions: latest; stable; 0.8.2; 0.8.1.post0; 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.