TextSearcher

publiczna klasa końcowa TextSearcher

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
  • Dla modelu TFLite opartego na Regex:
    • 1 tensor wejściowy.
    • input_process_units dla tokenizera RegexTokenizer
    • dokładnie jeden tensor wyjściowy typu kTfLiteFloat32
  • Dla modelu TFLite opartego na uniwersalnym koderze zdań:
    • 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
statyczna wyszukiwarka tekstu
statyczna wyszukiwarka tekstu
statyczna wyszukiwarka tekstu
createFromFileAndOptions (kontekst kontekstowy, String modelPath, opcje TextSearcher.TextSearcherOptions )
Tworzy instancję TextSearcher na podstawie TextSearcher.TextSearcherOptions .
Lista < 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.

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