Class that performs dense feature vector extraction on text.
tflite_support.task.text.TextEmbedder(
options: tflite_support.task.text.TextEmbedderOptions
,
cpp_embedder: _CppTextEmbedder
) -> None
Attributes |
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