Выполняет поиск по сходству в текстовой строке.
API ожидает модель TFLite с необязательными, но настоятельно рекомендуемыми метаданными модели TFLite. .
API ожидает модель TFLite с заполненными метаданными. Метаданные должны содержать следующую информацию:
- Для модели TFLite на основе Берта:
- 3 входных тензора типа kTfLiteString с именами «ids», «mask» и «segment_ids».
- input_process_units для токенизатора Wordpiece/Sentencepiece
- ровно один выходной тензор типа kTfLiteFloat32
- 1 входной тензор.
- input_process_units для токенизатора RegexTokenizer
- ровно один выходной тензор типа kTfLiteFloat32
- 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 | createFromBufferAndOptionsImpl (опции ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions , int indexFd) |
статический 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> поиск ( строковый текст)
Выполняет встраивание из предоставленной входной строки с последующим поиском ближайшего соседа по индексу.
Параметры
текст | ввести текстовый запрос к модели |
---|