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