TextSearcher

TextSearcher della classe finale pubblica

Esegue la ricerca per somiglianza sulla stringa di testo.

L'API prevede un modello TFLite con metadati del modello TFLite facoltativi, 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 Tokenizer 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 Tokenizzatore
    • esattamente un tensore di uscita di tipo kTfLiteFloat32
  • Per il modello TFLite basato su Universal Sentence Encoder:
    • 3 tensori di input con nomi "inp_text", "res_context" e "res_text"
    • 2 tensori di output con nomi "query_encoding" e "response_encoding" di tipo kTfLiteFloat32

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

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

Classi nidificate

classe TextSearcher.TextSearcherOptions Opzioni per impostare un TextSearcher.

Metodi pubblici

TextSearcher statico
createFromBufferAndOptions ( opzioni ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Crea un'istanza TextSearcher con un buffer del modello e TextSearcher.TextSearcherOptions .
TextSearcher statico
TextSearcher statico
TextSearcher statico
Elenco < Vicino più vicino >
ricerca (testo stringa )
Esegue l'estrazione dell'incorporamento sulla stringa di input fornita, seguita dalla ricerca del vicino più vicino nell'indice.

Metodi ereditati

Metodi pubblici

TextSearcher statico pubblico createFromBufferAndOptions (opzioni ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

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

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

TextSearcher statico pubblico createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opzioni TextSearcher.TextSearcherOptions , int indexFd)

Parametri
modelBuffer
opzioni
indiceFd

TextSearcher statico pubblico createFromFileAndOptions (opzioni File modelFile, TextSearcher.TextSearcherOptions )

Crea un'istanza TextSearcher .

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

textSearcher statico pubblico createFromFileAndOptions (contesto di contesto, string modelPath, opzioni TextSearcher.TextSearcherOptions )

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

elenco pubblico < vicino più vicino > ricerca (testo stringa )

Esegue l'estrazione dell'incorporamento sulla stringa di input fornita, seguita dalla ricerca del vicino più vicino nell'indice.

Parametri
testo inserire una query di testo nel modello