Shortcuts

SingleBinaryTargetFormatter

class flash.core.data.utilities.classification.SingleBinaryTargetFormatter(labels=None, num_classes=None)[source]

A TargetFormatter for targets that are one-hot encoded binaries.

Examples

>>> import torch
>>> from flash.core.data.utilities.classification import SingleBinaryTargetFormatter
>>> formatter = SingleBinaryTargetFormatter(num_classes=2)
>>> formatter([1, 0])
0
>>> formatter(torch.tensor([0, 1]))
1
Read the Docs v: latest
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.