Shortcuts

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...]