Wykonuje wyszukiwanie podobieństwa w ciągu tekstowym.
Interfejs API oczekuje modelu TFLite z opcjonalnymi, ale zdecydowanie zalecanymi metadanymi modelu TFLite. .
Interfejs API oczekuje modelu TFLite z wypełnionymi metadanymi. Metadane powinny zawierać następujące informacje:
- Dla modelu TFLite opartego na Bercie:
- 3 tensory wejściowe typu kTfLiteString o nazwach „ids”, „mask” i „segment_ids”.
- input_process_units dla tokenizera Wordpiece/Zdanie
- dokładnie jeden tensor wyjściowy typu kTfLiteFloat32
- 1 tensor wejściowy.
- input_process_units dla tokenizera RegexTokenizer
- dokładnie jeden tensor wyjściowy typu kTfLiteFloat32
- 3 tensory wejściowe o nazwach „inp_text”, „res_context” i „res_text”
- 2 tensory wyjściowe o nazwach „query_encoding” i „response_encoding” typu kTfLiteFloat32
DO ZROBIENIA(b/180502532): dodaj wskaźnik do przykładowego modelu.
DO ZROBIENIA (b/222671076): dodaj metody tworzenia fabryki bez opcji, takie jak „createFromFile”, gdy obsługiwany jest format pojedynczego pliku (plik indeksu spakowany w modelu).
Klasy zagnieżdżone
klasa | TextSearcher.TextSearcherOpcje | Opcje konfiguracji TextSearcher. |
Metody publiczne
statyczna wyszukiwarka tekstu | createFromBufferAndOptions (opcje ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions ) Tworzy instancję TextSearcher z buforem modelu i TextSearcher.TextSearcherOptions . |
statyczna wyszukiwarka tekstu | createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opcje TextSearcher.TextSearcherOptions , int indeksFd) |
statyczna wyszukiwarka tekstu | createFromFileAndOptions ( Opcje pliku modelFile, TextSearcher.TextSearcherOptions ) Tworzy instancję TextSearcher . |
statyczna wyszukiwarka tekstu | createFromFileAndOptions (kontekst kontekstowy, String modelPath, opcje TextSearcher.TextSearcherOptions ) Tworzy instancję TextSearcher na podstawie TextSearcher.TextSearcherOptions . |
Lista < Najbliższy Sąsiad > |
Metody dziedziczone
Metody publiczne
public static TextSearcher createFromBufferAndOptions (opcje ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Tworzy instancję TextSearcher
z buforem modelu i TextSearcher.TextSearcherOptions
.
Parametry
Bufor modelu | bezpośredni ByteBuffer lub MappedByteBuffer modelu wyszukiwania |
---|---|
opcje |
Rzuca
Wyjątek IllegalArgument | jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer |
---|---|
Wyjątek IO | jeśli podczas ładowania pliku indeksu wystąpi błąd we/wy |
Wyjątek IllegalStateException | jeśli wystąpił błąd wewnętrzny |
Wyjątek czasu wykonania | jeśli wystąpi inny nieokreślony błąd |
public static TextSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opcje TextSearcher.TextSearcherOptions , int indeksFd)
Parametry
Bufor modelu | |
---|---|
opcje | |
indeksFd |
public static TextSearcher createFromFileAndOptions (opcje pliku modelFile, TextSearcher.TextSearcherOptions )
Tworzy instancję TextSearcher
.
Parametry
plik modelu | model wyszukiwania Instancja File |
---|---|
opcje |
Rzuca
Wyjątek IO | jeśli wystąpi błąd we/wy podczas ładowania modelu tflite lub pliku indeksu |
---|---|
Wyjątek IllegalArgument | jeśli argument jest nieprawidłowy |
Wyjątek IllegalStateException | jeśli wystąpił błąd wewnętrzny |
Wyjątek czasu wykonania | jeśli wystąpi inny nieokreślony błąd |
public static TextSearcher createFromFileAndOptions (kontekst kontekstowy, String modelPath, opcje TextSearcher.TextSearcherOptions )
Tworzy instancję TextSearcher
na podstawie TextSearcher.TextSearcherOptions
.
Parametry
kontekst | |
---|---|
Ścieżka modelu | ścieżka modelu wyszukiwania z metadanymi w zasobach |
opcje |
Rzuca
Wyjątek IO | jeśli wystąpi błąd we/wy podczas ładowania modelu tflite lub pliku indeksu |
---|---|
Wyjątek IllegalArgument | jeśli argument jest nieprawidłowy |
Wyjątek IllegalStateException | jeśli wystąpił błąd wewnętrzny |
Wyjątek czasu wykonania | jeśli wystąpi inny nieokreślony błąd |
Lista publiczna < Najbliższy Sąsiad > szukaj ( ciąg tekstowy)
Wykonuje wyodrębnianie osadzania na podanym ciągu wejściowym, a następnie wyszukiwanie najbliższego sąsiada w indeksie.
Parametry
tekst | wprowadź zapytanie tekstowe do modelu |
---|