MediaPipeTasksText Framework Reference

Classifications

class Classifications : NSObject

Represents the list of classification for a given classifier head. Typically used as a result for classification tasks.

  • The index of the classifier head these entries refer to. This is useful for multi-head models.

    Declaration

    Swift

    var headIndex: Int { get }
  • The optional name of the classifier head, which is the corresponding tensor metadata name.

    Declaration

    Swift

    var headName: String? { get }
  • An array of Category objects containing the predicted categories.

    Declaration

    Swift

    var categories: [ResultCategory] { get }
  • Initializes a new Classifications object with the given head index and array of categories. Head name is initialized to nil.

    Declaration

    Swift

    convenience init(head headIndex: Int, categories: [ResultCategory])

    Parameters

    headIndex

    The index of the classifier head.

    categories

    An array of Category objects containing the predicted categories.

    Return Value

    An instance of Classifications initialized with the given head index and array of categories.

  • Initializes a new Classifications with the given head index, head name and array of categories.

    Declaration

    Swift

    init(head headIndex: Int, headName: String?, categories: [ResultCategory])

    Parameters

    headIndex

    The index of the classifier head.

    headName

    The name of the classifier head, which is the corresponding tensor metadata name.

    categories

    An array of Category objects containing the predicted categories.

    Return Value

    An object of Classifications initialized with the given head index, head name and array of categories.

  • Undocumented