TextSearcher

TextSearcher de clase final pública

Realiza búsqueda de similitud en cadenas de texto.

La API espera un modelo TFLite con metadatos del modelo TFLite opcionales, pero muy recomendados. .

La API espera un modelo TFLite con metadatos completos. 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 tokenizador de Wordpiece/Sentencepiece
    • 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 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): agrega 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 (opciones ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Crea una instancia TextSearcher con un búfer de modelo y TextSearcher.TextSearcherOptions .
Buscador de texto estático
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opciones TextSearcher.TextSearcherOptions , int indexFd)
Buscador de texto estático
createFromFileAndOptions (opciones de archivo modelFile, TextSearcher.TextSearcherOptions )
Crea una instancia TextSearcher .
Buscador de texto estático
createFromFileAndOptions (contexto contextual, ruta del modelo de cadena , opciones TextSearcher.TextSearcherOptions )
Crea una instancia TextSearcher a partir de TextSearcher.TextSearcherOptions .
Lista < Vecino más cercano >
buscar (texto de cadena )
Realiza una extracción de incrustación en la entrada de cadena proporcionada, seguida de una búsqueda del vecino más cercano en el índice.

Métodos heredados

Métodos públicos

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

Crea una instancia 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
IOExcepción si se produce un error de E/S al cargar el archivo de índice
Excepción de estado ilegal 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, opciones TextSearcher.TextSearcherOptions , int indexFd)

Parámetros
modeloBuffer
opciones
índiceFd

TextSearcher estático público createFromFileAndOptions (opciones de archivo modelFile, TextSearcher.TextSearcherOptions )

Crea una instancia TextSearcher .

Parámetros
modeloArchivo la instancia File del modelo de búsqueda
opciones
Lanza
IOExcepción 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
Excepción de estado ilegal 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 contextual, ruta del modelo de cadena , opciones TextSearcher.TextSearcherOptions )

Crea una instancia TextSearcher a partir de TextSearcher.TextSearcherOptions .

Parámetros
contexto
modeloRuta ruta del modelo de búsqueda con metadatos en los activos
opciones
Lanza
IOExcepción 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
Excepción de estado ilegal 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 > buscar (texto de cadena )

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

Parámetros
texto consulta de texto de entrada al modelo