TextSearcher

TextSearcher de clase final pública

Realiza una búsqueda de similitud en una cadena de texto.

La API espera un modelo TFLite con metadatos de modelo TFLite opcionales, pero muy recomendables . .

La API espera un modelo TFLite con metadatos poblados. Los metadatos deben contener la siguiente información:

  • Para el modelo TFLite basado en Bert:
    • 3 tensores de entrada de tipo kTfLiteString con nombres "ids", "mask" y "segment_ids".
    • input_process_units para Wordpiece/Sentencepiece Tokenizer
    • exactamente un tensor de salida de tipo kTfLiteFloat32
  • Para el modelo TFLite basado en Regex:
    • 1 tensor de entrada.
    • input_process_units para el tokenizador RegexTokenizer
    • exactamente un tensor de salida de tipo kTfLiteFloat32
  • Para el modelo TFLite basado en Universal Sentence Encoder:
    • 3 tensores de entrada con los nombres "inp_text", "res_context" y "res_text"
    • 2 tensores de salida con nombres "query_encoding" y "response_encoding" de tipo kTfLiteFloat32

TODO(b/180502532): agregue un puntero al modelo de ejemplo.

TODO(b/222671076): agregue métodos de creación de fábrica sin opciones, como `createFromFile`, una vez que se admita el formato de archivo único (archivo de índice empaquetado en el modelo).

Clases anidadas

clase TextSearcher.TextSearcherOptions Opciones para configurar un TextSearcher.

Métodos públicos

buscador de texto estático
createFromBufferAndOptions ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions opciones)
Crea una instancia de TextSearcher con un búfer de modelo y TextSearcher.TextSearcherOptions .
buscador de texto estático
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions opciones, int indexFd)
buscador de texto estático
createFromFileAndOptions (Modelo de archivoArchivo, TextSearcher.TextSearcherOptions opciones )
Crea una instancia de TextSearcher .
buscador de texto estático
createFromFileAndOptions (contexto contextual, String modelPath, TextSearcher.TextSearcherOptions opciones)
Crea una instancia de TextSearcher a partir de TextSearcher.TextSearcherOptions .
Lista < Vecino más cercano>
buscar ( cadena de texto)
Realiza la extracción incrustada en la entrada de cadena proporcionada, seguida de la búsqueda del vecino más cercano en el índice.

Métodos Heredados

Métodos públicos

TextSearcher estático público createFromBufferAndOptions ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions opciones)

Crea una instancia de TextSearcher con un búfer de modelo y TextSearcher.TextSearcherOptions .

Parámetros
modeloBuffer un ByteBuffer directo o un MappedByteBuffer del modelo de búsqueda
opciones
Lanza
Argumento de excepción ilegal si el búfer del modelo no es un ByteBuffer directo o un MappedByteBuffer
IOException si se produce un error de E/S al cargar el archivo de índice
IllegalStateExceptionIlegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

TextSearcher estático público createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions opciones, int indexFd)

Parámetros
modeloBuffer
opciones
índiceFd

TextSearcher estático público createFromFileAndOptions (Modelo de archivoFile , TextSearcher.TextSearcherOptions opciones)

Crea una instancia de TextSearcher .

Parámetros
modeloArchivo el modelo de búsqueda Instancia de File
opciones
Lanza
IOException si se produce un error de E/S al cargar el modelo tflite o el archivo de índice
Argumento de excepción ilegal si un argumento no es válido
IllegalStateExceptionIlegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

TextSearcher estático público createFromFileAndOptions (contexto de contexto, String modelPath, opciones de TextSearcher.TextSearcherOptions )

Crea una instancia de TextSearcher a partir de TextSearcher.TextSearcherOptions .

Parámetros
contexto
modeloPath ruta del modelo de búsqueda con metadatos en los activos
opciones
Lanza
IOException si se produce un error de E/S al cargar el modelo tflite o el archivo de índice
Argumento de excepción ilegal si un argumento no es válido
IllegalStateExceptionIlegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

lista pública < vecino más cercano> búsqueda (texto de cadena )

Realiza la extracción incrustada en la entrada de cadena proporcionada, seguida de la búsqueda del vecino más cercano en el índice.

Parámetros
texto consulta de texto de entrada al modelo