TextSearcher

classe finale pubblica TextSearcher

Esegue la ricerca di somiglianza sulla stringa di testo.

L'API prevede un modello TFLite con metadati del modello TFLite opzionali, ma fortemente consigliati . .

L'API prevede un modello TFLite con metadati popolati. I metadati dovrebbero contenere le seguenti informazioni:

  • Per il modello TFLite basato su Bert:
    • 3 tensori di input di tipo kTfLiteString con nomi "ids", "mask" e "segment_ids".
    • input_process_units per il token di Wordpiece/Sentencepiece
    • esattamente un tensore di uscita di tipo kTfLiteFloat32
  • Per il modello TFLite basato su Regex:
    • 1 tensore di ingresso.
    • input_process_units per RegexTokenizer Tokenizer
    • esattamente un tensore di uscita di tipo kTfLiteFloat32
  • Per il modello TFLite basato su codificatore di frasi universali:
    • 3 tensori di input con i nomi "inp_text", "res_context" e "res_text"
    • 2 tensori di uscita con i nomi "query_encoding" e "response_encoding" di tipo kTfLiteFloat32

TODO(b/180502532): aggiungi puntatore al modello di esempio.

TODO(b/222671076): aggiungi metodi di creazione di fabbrica senza opzioni, come `createFromFile`, una volta che il formato di file singolo (file di indice compresso nel modello) è supportato.

Classi nidificate

classe TextSearcher.TextSearcherOptions Opzioni per impostare un TextSearcher.

Metodi pubblici

TextSearcher statico
createFromBufferAndOptions ( ByteBuffer modelBuffer, opzioni TextSearcher.TextSearcherOptions )
Crea un'istanza di TextSearcher con un buffer del modello e TextSearcher.TextSearcherOptions .
TextSearcher statico
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opzioni TextSearcher.TextSearcherOptions , int indexFd)
TextSearcher statico
createFromFileAndOptions ( file modelFile, opzioni TextSearcher.TextSearcherOptions )
Crea un'istanza di TextSearcher .
TextSearcher statico
createFromFileAndOptions (contesto di contesto, string modelPath, opzioni TextSearcher.TextSearcherOptions )
Crea un'istanza di TextSearcher da TextSearcher.TextSearcherOptions .
Elenco < Vicino più vicino >
cerca (testo stringa )
Esegue l'estrazione di incorporamento sull'input di stringa fornito, seguita dalla ricerca del vicino più vicino nell'indice.

Metodi ereditati

Metodi pubblici

public static TextSearcher createFromBufferAndOptions ( opzioni ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Crea un'istanza di TextSearcher con un buffer del modello e TextSearcher.TextSearcherOptions .

Parametri
modello Buffer un ByteBuffer diretto o un MappedByteBuffer del modello di ricerca
opzioni
Tiri
IllegalArgumentException se il buffer del modello non è un ByteBuffer diretto o un MappedByteBuffer
IOException se si verifica un errore di I/O durante il caricamento del file di indice
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static TextSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opzioni TextSearcher.TextSearcherOptions , int indexFd)

Parametri
modello Buffer
opzioni
indiceFd

public static TextSearcher createFromFileAndOptions ( file modelFile, opzioni TextSearcher.TextSearcherOptions )

Crea un'istanza di TextSearcher .

Parametri
modelloFile il modello di ricerca Istanza del File
opzioni
Tiri
IOException se si verifica un errore di I/O durante il caricamento del modello tflite o del file di indice
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static TextSearcher createFromFileAndOptions (contesto di contesto, String modelPath, opzioni TextSearcher.TextSearcherOptions )

Crea un'istanza di TextSearcher da TextSearcher.TextSearcherOptions .

Parametri
contesto
modelPath percorso del modello di ricerca con metadati negli asset
opzioni
Tiri
IOException se si verifica un errore di I/O durante il caricamento del modello tflite o del file di indice
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public List < NearestNeighbor > cerca ( String text)

Esegue l'estrazione di incorporamento sull'input di stringa fornito, seguita dalla ricerca del vicino più vicino nell'indice.

Parametri
testo inserire una query di testo nel modello