Represents the classification results for a given classifier head.
mp.tasks.components.containers.Classifications(
categories: List[mp.tasks.components.containers.Category
],
head_index: int,
head_name: Optional[str] = None
)
Attributes |
categories
|
The array of predicted categories, usually sorted by descending
scores (e.g. from high to low probability).
|
head_index
|
The index of the classifier head these categories refer to. This
is useful for multi-head models.
|
head_name
|
The name of the classifier head, which is the corresponding
tensor metadata name.
|
Methods
__eq__
__eq__(
other
)
Class Variables |
head_name
|
None
|