tflite_support.task.text.TextEmbedder

Class that performs dense feature vector extraction on text.

tflite_support.task.text.TextEmbedder(
    options: tflite_support.task.text.TextEmbedderOptions,
    cpp_embedder: _CppTextEmbedder
) -> None

number_of_output_layers Gets the number of output layers of the model.
options

Methods

cosine_similarity

View source

cosine_similarity(
    u: tflite_support.task.processor.FeatureVector,
    v: tflite_support.task.processor.FeatureVector
) -> float

Computes cosine similarity [1] between two feature vectors.

create_from_file

View source

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

Creates the TextEmbedder object from a TensorFlow Lite model.

Args
file_path Path to the model.

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

Raises
ValueError If failed to create TextEmbedder object from the provided file such as invalid file.
RuntimeError If other types of error occurred.

create_from_options