MultiSoftTargetFormatter¶
- class flash.core.data.utilities.classification.MultiSoftTargetFormatter(labels=None, num_classes=None)[source]¶
A
TargetFormatter
for mutli-label soft targets.Examples
>>> import torch >>> from flash.core.data.utilities.classification import MultiSoftTargetFormatter >>> formatter = MultiSoftTargetFormatter(num_classes=3) >>> formatter([0.1, 0.9, 0.6]) [0.1, 0.9, 0.6] >>> formatter(torch.tensor([0.9, 0.6, 0.7])) [0..., 0..., 0...]