TextSearcher

публичный финальный класс TextSearcher

Выполняет поиск по сходству в текстовой строке.

API ожидает модель TFLite с необязательными, но настоятельно рекомендуемыми метаданными модели TFLite. .

API ожидает модель TFLite с заполненными метаданными. Метаданные должны содержать следующую информацию:

  • Для модели TFLite на основе Берта:
    • 3 входных тензора типа kTfLiteString с именами «ids», «mask» и «segment_ids».
    • input_process_units для токенизатора Wordpiece/Sentencepiece
    • ровно один выходной тензор типа kTfLiteFloat32
  • Для модели TFLite на основе Regex:
    • 1 входной тензор.
    • input_process_units для токенизатора RegexTokenizer
    • ровно один выходной тензор типа kTfLiteFloat32
  • Для модели TFLite на основе универсального кодировщика предложений:
    • 3 входных тензора с именами «inp_text», «res_context» и «res_text».
    • 2 выходных тензора с именами «query_encoding» и «response_encoding» типа kTfLiteFloat32.

TODO(b/180502532): добавить указатель на пример модели.

TODO(b/222671076): добавить заводские методы создания без опций, такие как createFromFile, как только будет поддерживаться единый формат файла (индексный файл, упакованный в модель).

Вложенные классы

сорт TextSearcher.TextSearcherOptions Варианты настройки TextSearcher.

Публичные методы

статический TextSearcher
createFromBufferAndOptions (опции ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Создает экземпляр TextSearcher с буфером модели и TextSearcher.TextSearcherOptions .
статический TextSearcher
статический TextSearcher
createFromFileAndOptions (опции File modelFile, TextSearcher.TextSearcherOptions )
Создает экземпляр TextSearcher .
статический TextSearcher
createFromFileAndOptions (контекст контекста, String modelPath, параметры TextSearcher.TextSearcherOptions )
Создает экземпляр TextSearcher из TextSearcher.TextSearcherOptions .
Список < ближайший сосед >
поиск (текстовая строка )
Выполняет встраивание из предоставленной входной строки с последующим поиском ближайшего соседа по индексу.

Унаследованные методы

Публичные методы

общедоступный статический TextSearcher createFromBufferAndOptions (параметры ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Создает экземпляр TextSearcher с буфером модели и TextSearcher.TextSearcherOptions .

Параметры
модельBuffer прямой ByteBuffer или MappedByteBuffer модели поиска
параметры
Броски
IllegalArgumentException если буфер модели не является прямым ByteBuffer или MappedByteBuffer
Исключение IO если при загрузке индексного файла возникает ошибка ввода-вывода
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

public static TextSearcher createFromBufferAndOptionsImpl (опции ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions , int indexFd)

Параметры
модельBuffer
параметры
indexFd

public static TextSearcher createFromFileAndOptions (опции File modelFile, TextSearcher.TextSearcherOptions )

Создает экземпляр TextSearcher .

Параметры
МодельФайл экземпляр File модели поиска
параметры
Броски
Исключение IO если возникает ошибка ввода-вывода при загрузке модели tflite или индексного файла
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

public static TextSearcher createFromFileAndOptions (контекст контекста, String modelPath, параметры TextSearcher.TextSearcherOptions )

Создает экземпляр TextSearcher из TextSearcher.TextSearcherOptions .

Параметры
контекст
модельпуть путь модели поиска с метаданными в активах
параметры
Броски
Исключение IO если возникает ошибка ввода-вывода при загрузке модели tflite или индексного файла
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

общедоступный список <NearestNeighbor> поиск ( строковый текст)

Выполняет встраивание из предоставленной входной строки с последующим поиском ближайшего соседа по индексу.

Параметры
текст ввести текстовый запрос к модели