Class that performs NL classification on text.
tflite_support
.
task
.
text
.
NLClassifier
(
options
:
tflite_support
.
task
.
text
.
NLClassifierOptions
,
cpp_classifier
:
_CppNLClassifier
)
->
None
Methods
classify
classify
(
text
:
str
)
->
tflite_support
.
task
.
processor
.
ClassificationResult
Performs actual NL classification on the provided text.
Returns
The classification result.
Raises
ValueError
If any of the input arguments is invalid.
RuntimeError
If failed to perform the classification.
create_from_file
@classmethodcreate_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.
create_from_options
@classmethodcreate_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.
