tflite_support.task.text.NLClassifier

Class that performs NL classification on text.

options

Methods

classify

View source

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.

create_from_file

View source

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.

create_from_options

View source

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.