List of predicted classes (aka labels) for a given classifier head.

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.



View source

Checks if this object is equal to the given object.

other The object to be compared with.

True if the objects are equal.