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]