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
classify
(
text
:
str
)
->
tflite_support
.
task
.
processor
.
ClassificationResult
Performs actual Bert NL classification on the provided text.
Returns
classification result.
Raises
ValueError
If any of the input arguments is invalid.
RuntimeError
If failed to calculate the embedding vector.
create_from_file
@classmethodcreate_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
@classmethodcreate_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.
