SingleNumericTargetFormatter¶
- class flash.core.data.utilities.classification.SingleNumericTargetFormatter(labels=None, num_classes=None)[source]¶
A
TargetFormatter
for targets that contain a single numeric value (the class index).Examples
>>> import torch >>> from flash.core.data.utilities.classification import SingleNumericTargetFormatter >>> formatter = SingleNumericTargetFormatter(num_classes=10) >>> formatter(5) 5 >>> formatter([5]) 5 >>> formatter(torch.tensor(5)) 5