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]