MultiNumericTargetFormatter¶

class flash.core.data.utilities.classification.MultiNumericTargetFormatter(labels=None, num_classes=None)[source]¶

A TargetFormatter for targets that contain multiple numeric values (the class indices).

Examples

>>> import torch
>>> from flash.core.data.utilities.classification import MultiNumericTargetFormatter
>>> formatter = MultiNumericTargetFormatter(num_classes=10)
>>> formatter([2, 5])
[0, 0, 1, 0, 0, 1, 0, 0, 0, 0]
>>> formatter(torch.tensor([2, 5]))
[0, 0, 1, 0, 0, 1, 0, 0, 0, 0]