SpaceDelimitedTargetFormatter¶

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

A TargetFormatter for targets that contain a string with multiple space-delimited labels.

Examples

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