TextSearcher

classe final pública TextSearcher

Executa pesquisa de similaridade em string de texto.

A API espera um modelo TFLite com metadados de modelo TFLite opcionais, mas altamente recomendados. .

A API espera um modelo TFLite com metadados preenchidos. Os metadados devem conter as seguintes informações:

  • Para o modelo TFLite baseado em Bert:
    • 3 tensores de entrada do tipo kTfLiteString com nomes "ids", "mask" e "segment_ids".
    • input_process_units para tokenizador de peça de palavra/frase
    • exatamente um tensor de saída do tipo kTfLiteFloat32
  • Para modelo TFLite baseado em Regex:
    • 1 tensor de entrada.
    • input_process_units para tokenizador RegexTokenizer
    • exatamente um tensor de saída do tipo kTfLiteFloat32
  • Para modelo TFLite baseado em codificador de frase universal:
    • 3 tensores de entrada com nomes "inp_text", "res_context" e "res_text"
    • 2 tensores de saída com nomes "query_encoding" e "response_encoding" do tipo kTfLiteFloat32

TODO(b/180502532): adiciona ponteiro ao modelo de exemplo.

TODO(b/222671076): adiciona métodos de criação de fábrica sem opções, como `createFromFile`, uma vez que o formato de arquivo único (arquivo de índice compactado no modelo) seja suportado.

Classes aninhadas

aula TextSearcher.TextSearcherOptions Opções para configurar um TextSearcher.

Métodos Públicos

Pesquisador de texto estático
createFromBufferAndOptions (opções ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Cria uma instância TextSearcher com um buffer de modelo e TextSearcher.TextSearcherOptions .
Pesquisador de texto estático
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opções TextSearcher.TextSearcherOptions , int indexFd)
Pesquisador de texto estático
createFromFileAndOptions (opções de arquivo modelFile, TextSearcher.TextSearcherOptions )
Cria uma instância TextSearcher .
Pesquisador de texto estático
createFromFileAndOptions (contexto de contexto, String modelPath, opções TextSearcher.TextSearcherOptions )
Cria uma instância TextSearcher de TextSearcher.TextSearcherOptions .
Lista < Vizinho mais próximo >
pesquisa (texto de string )
Executa a extração de incorporação na entrada de string fornecida, seguida pela pesquisa do vizinho mais próximo no índice.

Métodos herdados

Métodos Públicos

public static TextSearcher createFromBufferAndOptions (opções ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Cria uma instância TextSearcher com um buffer de modelo e TextSearcher.TextSearcherOptions .

Parâmetros
modeloBuffer um ByteBuffer direto ou um MappedByteBuffer do modelo de pesquisa
opções
Lança
Exceção de argumento ilegal se o buffer do modelo não for um ByteBuffer direto ou um MappedByteBuffer
IOException se ocorrer um erro de E/S ao carregar o arquivo de índice
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

public static TextSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opções TextSearcher.TextSearcherOptions , int indexFd)

Parâmetros
modeloBuffer
opções
índiceFd

public static TextSearcher createFromFileAndOptions (opções de arquivo modelFile, TextSearcher.TextSearcherOptions )

Cria uma instância TextSearcher .

Parâmetros
arquivomodelo a instância File do modelo de pesquisa
opções
Lança
IOException se ocorrer um erro de E/S ao carregar o modelo tflite ou o arquivo de índice
Exceção de argumento ilegal se um argumento for inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

public static TextSearcher createFromFileAndOptions (contexto de contexto, String modelPath, opções TextSearcher.TextSearcherOptions )

Cria uma instância TextSearcher de TextSearcher.TextSearcherOptions .

Parâmetros
contexto
caminho do modelo caminho do modelo de pesquisa com metadados nos ativos
opções
Lança
IOException se ocorrer um erro de E/S ao carregar o modelo tflite ou o arquivo de índice
Exceção de argumento ilegal se um argumento for inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

lista pública < vizinho mais próximo > pesquisa (texto de string )

Executa a extração de incorporação na entrada de string fornecida, seguida pela pesquisa do vizinho mais próximo no índice.

Parâmetros
texto consulta de texto de entrada para o modelo