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