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 tonil
.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
-