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