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