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]