MultiLabelTargetFormatter¶

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

A TargetFormatter for targets that contain multiple string labels in a list.

Examples

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