Introducing Google AI Edge Portal: Benchmark Edge AI at scale. Sign-up to request access during private preview.

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.