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]