tflite_support.task.text.BertCluAnnotator

Class that performs Bert CLU Annotation on text.

tflite_support.task.text.BertCluAnnotator(
    options: tflite_support.task.text.BertCluAnnotatorOptions,
    cpp_annotator: _CppBertCluAnnotator
) -> None

options

Methods

annotate

View source

annotate(
    request: tflite_support.task.processor.CluRequest
) -> tflite_support.task.processor.CluResponse

Performs actual Bert CLU Annotation on the provided CLU request.

Args
requestThe input to CLU.

Returns
The output of CLU.

Raises
ValueErrorIf any of the input arguments is invalid.
RuntimeErrorIf failed to calculate the embedding vector.

create_from_file

View source

@classmethod
create_from_file(
    file_path: str
) -> 'BertCluAnnotator'

Creates the BertCluAnnotator object from a TensorFlow Lite model.

Args
file_pathPath to the model.

Returns
BertCLUAnnotator object that's created from the model file.

Raises
ValueErrorIf failed to create BertCluAnnotator object from the provided file such as invalid file.
RuntimeErrorIf other types of error occurred.

create_from_options

View source

@classmethod
create_from_options(
    options: tflite_support.task.text.BertCluAnnotatorOptions
) -> 'BertCluAnnotator'

Creates the BertCluAnnotator object from Bert CLU Annotation options.

Args
optionsOptions for the Bert CLU annotation task.

Returns
BertCluAnnotator object that's created from options.

Raises
ValueErrorIf failed to create BertCluAnnotator object from BertCluAnnotatorOptions such as missing the model.
RuntimeErrorIf other types of error occurred.