Introducing LiteRT: Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite. Learn more

tflite_support.task.text.NLClassifier

Class that performs NL classification on text.

tflite_support.task.text.NLClassifier(
    options: tflite_support.task.text.NLClassifierOptions,
    cpp_classifier: _CppNLClassifier
) -> None

Attributes
`options`

Attributes

options

classify(
    text: str
) -> tflite_support.task.processor.ClassificationResult

Performs actual NL classification on the provided text.

Args
`text`	the input text, used to extract the feature vectors.

Returns
The classification result.

Raises
`ValueError`	If any of the input arguments is invalid.
`RuntimeError`	If failed to perform the classification.

@classmethod
create_from_file(
    file_path: str
) -> 'NLClassifier'

Creates the NLClassifier object from a TensorFlow Lite model.

Args
`file_path`	Path to the model.

Returns
`NLClassifier` object that's created from the model file.

Raises
`ValueError`	If failed to create `NLClassifier` object from the provided file such as invalid file.
`RuntimeError`	If other types of error occurred.

@classmethod
create_from_options(
    options: tflite_support.task.text.NLClassifierOptions
) -> 'NLClassifier'

Creates the NLClassifier object from NL classifier options.

Args
`options`	Options for the NL classifier task.

Returns
`NLClassifier` object that's created from `options`.

Raises
`ValueError`	If failed to create `NLClassifier` object from `NLClassifierOptions` such as missing the model or if any of the classification options is invalid.
`RuntimeError`	If other types of error occurred.