tflite_support.task.text.BertNLClassifier

Class that performs Bert NL classification on text.

tflite_support.task.text.BertNLClassifier(
    options: tflite_support.task.text.BertNLClassifierOptions,
    cpp_classifier: _CppBertNLClassifier
) -> None

options

Methods

classify

View source

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

Performs actual Bert NL classification on the provided text.

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

Returns
classification result.

Raises
ValueError If any of the input arguments is invalid.
RuntimeError If failed to calculate the embedding vector.

create_from_file

View source

@classmethod
create_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

View source

@classmethod
create_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.