Introducing LiteRT: Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
tflite_support.task.text.BertNLClassifier
Stay organized with collections
Save and categorize content based on your preferences.
Class that performs Bert NL classification on text.
tflite_support.task.text.BertNLClassifier(
options: tflite_support.task.text.BertNLClassifierOptions
,
cpp_classifier: _CppBertNLClassifier
) -> None
Methods
classify
View source
classify(
text: str
) -> tflite_support.task.processor.ClassificationResult
Performs actual Bert NL classification on the provided text.
Args |
text
|
the input text, used to extract the feature vectors.
|
Returns |
classification result.
|
Raises |
ValueError
|
If any of the input arguments is invalid.
|
RuntimeError
|
If failed to calculate the embedding vector.
|
create_from_file
View source
@classmethod
create_from_file(
file_path: str
) -> 'BertNLClassifier'
Creates the BertNLClassifier
object from a TensorFlow Lite model.
Args |
file_path
|
Path to the model.
|
Returns |
BertNLClassifier object that's created from the model file.
|
Raises |
ValueError
|
If failed to create BertNLClassifier object from the
provided file such as invalid file.
|
RuntimeError
|
If other types of error occurred.
|
create_from_options
View source
@classmethod
create_from_options(
options: tflite_support.task.text.BertNLClassifierOptions
) -> 'BertNLClassifier'
Creates the BertNLClassifier
object from Bert NL classifier options.
Args |
options
|
Options for the Bert NL classifier task.
|
Returns |
BertNLClassifier object that's created from options .
|
Raises |
ValueError
|
If failed to create BertNLClassifier object from
BertNLClassifierOptions such as missing the model.
|
RuntimeError
|
If other types of error occurred.
|
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"
}]
{"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."],[],[]]