Shortcuts

MultiBinaryTargetFormatter

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

A TargetFormatter for targets that are multi-hot binary.

Examples

>>> import torch
>>> from flash.core.data.utilities.classification import MultiBinaryTargetFormatter
>>> formatter = MultiBinaryTargetFormatter(num_classes=3)
>>> formatter([0, 1, 1])
[0, 1, 1]
>>> formatter(torch.tensor([1, 0, 0]))
[1, 0, 0]
Read the Docs v: 0.7.1
Versions
latest
stable
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
docs-fix_typing
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.