Introducing LiteRT : Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
Send feedback
tflite_support.task.processor.ClassificationOptions
Stay organized with collections
Save and categorize content based on your preferences.
Options for classification processor.
tflite_support . task . processor . ClassificationOptions (
score_threshold : Optional [ float ] = None ,
category_name_allowlist : Optional [ List [ str ]] = None ,
category_name_denylist : Optional [ List [ str ]] = None ,
display_names_locale : Optional [ str ] = None ,
max_results : Optional [ int ] = None
)
Attributes
display_names_locale
The locale to use for display names specified through
the TFLite Model Metadata.
max_results
The maximum number of top-scored classification results to
return.
score_threshold
Overrides the ones provided in the model metadata. Results
below this value are rejected.
category_name_allowlist
If non-empty, classifications whose class name is
not in this set will be filtered out. Duplicate or unknown class names are
ignored. Mutually exclusive with category_name_denylist
.
category_name_denylist
If non-empty, classifications whose class name is in
this set will be filtered out. Duplicate or unknown class names are
ignored. Mutually exclusive with category_name_allowlist
.
Methods
create_from_pb2
View source
@classmethod
create_from_pb2 (
pb2_obj : _ClassificationOptionsProto
) -> 'ClassificationOptions'
Creates a ClassificationOptions
object from the given protobuf object.
to_pb2
View source
to_pb2 () -> _ClassificationOptionsProto
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.
Class Variables
category_name_allowlist
None
category_name_denylist
None
display_names_locale
None
max_results
None
score_threshold
None
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.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
Need to tell us more?
{"lastModified": "Last updated 2024-05-08 UTC."}
[[["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."],[],[]]