TextSearcher

classe final pública TextSearcher

Executa a pesquisa de similaridade na 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 Wordpiece/Sentencepiece Tokenizer
    • exatamente um tensor de saída do tipo kTfLiteFloat32
  • Para o modelo TFLite baseado em Regex:
    • 1 tensor de entrada.
    • input_process_units para RegexTokenizer Tokenizer
    • exatamente um tensor de saída do tipo kTfLiteFloat32
  • Para o modelo TFLite baseado no Universal Sentence Encoder:
    • 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): adicionar 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) é suportado.

Classes aninhadas

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

Métodos públicos

Pesquisador de Texto estático
createFromBufferAndOptions ( ByteBuffer modelBuffer, opções TextSearcher.TextSearcherOptions )
Cria uma instância TextSearcher com um buffer de modelo e TextSearcher.TextSearcherOptions .
Pesquisador de Texto estático
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions options, 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
indexFd

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

Cria uma instância TextSearcher .

Parâmetros
modelFile 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
modelPath 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

public List < NearestNeighbor > search ( String text)

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