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
request The input to CLU.

Returns
The output of CLU.

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
) -> 'BertCluAnnotator'

Creates the BertCluAnnotator object from a TensorFlow Lite model.

Args
file_path Path to the model.

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

Raises
ValueError If failed to create BertCluAnnotator 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.BertCluAnnotatorOptions
) -> 'BertCluAnnotator'

Creates the BertCluAnnotator object from Bert CLU Annotation options.

Args
options Options for the Bert CLU annotation task.

Returns
BertCluAnnotator object that's created from options.

Raises
ValueError If failed to create BertCluAnnotator object from BertCluAnnotatorOptions such as missing the model.
RuntimeError If other types of error occurred.