API-интерфейс классификатора для задач NLClassification с моделями Берта классифицирует строки по различным классам. API ожидает модель TFLite на основе Берта с заполненными метаданными.
Метаданные должны содержать следующую информацию:
- 1 input_process_unit для токенизатора слов/предложений.
- 3 входных тензора с именами «ids», «mask» и «segment_ids».
- 1 выходной тензор типа float32[1, 2] с опционально прикрепленным файлом метки. Если файл метки прикреплен, он должен представлять собой обычный текстовый файл с одной меткой в строке. Количество меток должно соответствовать количеству категорий, выдаваемых моделью.
Вложенные классы
сорт | BertNLClassifier.BertNLClassifierOptions | Параметры настройки BertNLClassifier. |
Публичные методы
Унаследованные методы
Публичные методы
общедоступный список < Категория > классифицировать ( строковый текст)
Выполняет классификацию строкового ввода, возвращает классифицированные Category
.
Параметры
текст | введите текст в модель. |
---|
Возврат
- Список результатов по категориям.
общедоступный статический BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)
Создает BertNLClassifier
с буфером модели и BertNLClassifier.BertNLClassifierOptions
по умолчанию.
Параметры
модельBuffer | прямой ByteBuffer или MappedByteBuffer модели |
---|
Возврат
- экземпляр
BertNLClassifier
Броски
IllegalArgumentException | если буфер модели не является прямым ByteBuffer или MappedByteBuffer |
---|---|
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический BertNLClassifier createFromBufferAndOptions (опции ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Создает BertNLClassifier
с буфером модели и BertNLClassifier.BertNLClassifierOptions
.
Параметры
модельBuffer | прямой ByteBuffer или MappedByteBuffer модели |
---|---|
параметры | настроить классификатор |
Возврат
- экземпляр
BertNLClassifier
Броски
IllegalArgumentException | если буфер модели не является прямым ByteBuffer или MappedByteBuffer |
---|---|
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический BertNLClassifier createFromFile (контекстный контекст, String modelPath)
Создает BertNLClassifier
из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions
по умолчанию.
Параметры
контекст | Контекст Android |
---|---|
модельпуть | Путь к модели классификации |
Возврат
- экземпляр
BertNLClassifier
Броски
Исключение IO | Если файл модели не загружается |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
public static BertNLClassifier createFromFile ( Файл modelFile)
Создает BertNLClassifier
из объекта File
с метаданными и BertNLClassifier.BertNLClassifierOptions
по умолчанию.
Параметры
МодельФайл | Экземпляр File модели классификации |
---|
Возврат
- экземпляр
BertNLClassifier
Броски
Исключение IO | Если файл модели не загружается |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
public static BertNLClassifier createFromFileAndOptions (опции File modelFile, BertNLClassifier.BertNLClassifierOptions )
Создает BertNLClassifier
из объекта File
с метаданными и BertNLClassifier.BertNLClassifierOptions
.
Параметры
МодельФайл | Экземпляр File модели классификации |
---|---|
параметры | настроить классификатор |
Возврат
- экземпляр
BertNLClassifier
Броски
Исключение IO | Если файл модели не загружается |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |
общедоступный статический BertNLClassifier createFromFileAndOptions (контекст контекста, String modelPath, параметры BertNLClassifier.BertNLClassifierOptions )
Создает BertNLClassifier
из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions
.
Параметры
контекст | Контекст Android. |
---|---|
модельпуть | Путь к модели классификации |
параметры | настроить классификатор |
Возврат
- экземпляр
BertNLClassifier
Броски
Исключение IO | Если файл модели не загружается |
---|---|
IllegalArgumentException | если аргумент недействителен |
IllegalStateException | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |