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
- 1 tensore di ingresso.
- input_process_units per RegexTokenizer Tokenizzatore
- esattamente un tensore di uscita di tipo kTfLiteFloat32
- 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 | createFromBufferAndOptionsImpl ( opzioni ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions , int indexFd) |
TextSearcher statico | createFromFileAndOptions (opzioni File modelFile, TextSearcher.TextSearcherOptions ) Crea un'istanza TextSearcher . |
TextSearcher statico | createFromFileAndOptions (contesto di contesto, string modelPath, opzioni TextSearcher.TextSearcherOptions ) Crea un'istanza TextSearcher da TextSearcher.TextSearcherOptions . |
Elenco < Vicino più vicino > |
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 )
Crea un'istanza TextSearcher
da 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 |
---|