Introducing LiteRT : Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
Send feedback
tflite_support.task.processor.Classifications
List of predicted classes (aka labels) for a given classifier head.
tflite_support . task . processor . Classifications (
categories : List [ tflite_support . task . processor . Category
],
head_index : int ,
head_name : str
)
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.
Methods
create_from_pb2
View source
@classmethod
create_from_pb2 (
pb2_obj : _ClassificationsProto
) -> 'Classifications'
Creates a Classifications
object from the given protobuf object.
to_pb2
View source
to_pb2 () -> _ClassificationsProto
Generates a protobuf object to pass to the C++ layer.
__eq__
View source
__eq__ (
other : Any
) -> bool
Checks if this object is equal to the given object.
Args
other
The object to be compared with.
Returns
True if the objects are equal.
Send feedback
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License , and code samples are licensed under the Apache 2.0 License . For details, see the Google Developers Site Policies . Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-05-08 UTC.
Need to tell us more?
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-05-08 UTC."],[],[]]