BertNLClassifier

общедоступный класс BertNLClassifier

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 по умолчанию.
статический BertNLClassifier
статический BertNLClassifier
createFromFile (контекстный контекст, String modelPath)
Создает BertNLClassifier из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions по умолчанию.
статический BertNLClassifier
createFromFile ( Файл модели файла)
Создает BertNLClassifier из объекта File с метаданными и BertNLClassifier.BertNLClassifierOptions по умолчанию.
статический BertNLClassifier
createFromFileAndOptions (опции File modelFile, BertNLClassifier.BertNLClassifierOptions )
Создает BertNLClassifier из объекта File с метаданными и BertNLClassifier.BertNLClassifierOptions .
статический BertNLClassifier
createFromFileAndOptions (контекст контекста, String modelPath, параметры BertNLClassifier.BertNLClassifierOptions )
Создает BertNLClassifier из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions .

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

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

общедоступный список < Категория > классифицировать ( строковый текст)

Выполняет классификацию строкового ввода, возвращает классифицированные Category .

Параметры
текст введите текст в модель.
Возврат
  • Список результатов по категориям.

общедоступный статический BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Создает BertNLClassifier с буфером модели и BertNLClassifier.BertNLClassifierOptions по умолчанию.

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

общедоступный статический BertNLClassifier createFromBufferAndOptions (опции ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

Создает BertNLClassifier с буфером модели и BertNLClassifier.BertNLClassifierOptions .

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

общедоступный статический BertNLClassifier createFromFile (контекстный контекст, String modelPath)

Создает BertNLClassifier из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions по умолчанию.

Параметры
контекст Контекст Android
модельпуть Путь к модели классификации
Возврат
Броски
Исключение IO Если файл модели не загружается
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка

public static BertNLClassifier createFromFile ( Файл modelFile)

Создает BertNLClassifier из объекта File с метаданными и BertNLClassifier.BertNLClassifierOptions по умолчанию.

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

public static BertNLClassifier createFromFileAndOptions (опции File modelFile, BertNLClassifier.BertNLClassifierOptions )

Создает BertNLClassifier из объекта File с метаданными и BertNLClassifier.BertNLClassifierOptions .

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

общедоступный статический BertNLClassifier createFromFileAndOptions (контекст контекста, String modelPath, параметры BertNLClassifier.BertNLClassifierOptions )

Создает BertNLClassifier из файла модели с метаданными и BertNLClassifier.BertNLClassifierOptions .

Параметры
контекст Контекст Android.
модельпуть Путь к модели классификации
параметры настроить классификатор
Возврат
Броски
Исключение IO Если файл модели не загружается
IllegalArgumentException если аргумент недействителен
IllegalStateException если произошла внутренняя ошибка
Исключение времени выполнения если есть неуказанная ошибка