CommaDelimitedMultiLabelTargetFormatter¶
- class flash.core.data.utilities.classification.CommaDelimitedMultiLabelTargetFormatter(labels=None, num_classes=None)[source]¶
A
TargetFormatter
for targets that contain a string with multiple comma-delimited labels.Examples
>>> from flash.core.data.utilities.classification import CommaDelimitedMultiLabelTargetFormatter >>> formatter = CommaDelimitedMultiLabelTargetFormatter(labels=["bird", "cat", "dog"], num_classes=3) >>> formatter("cat,dog") [0, 1, 1] >>> formatter("bird") [1, 0, 0]