SingleLabelTargetFormatter¶

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

A TargetFormatter for targets that contain a single string label.

Examples

>>> from flash.core.data.utilities.classification import SingleLabelTargetFormatter
>>> formatter = SingleLabelTargetFormatter(labels=["cat", "dog"], num_classes=2)
>>> formatter("cat")
0
>>> formatter(["dog"])
1