NLClassifier

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

API-интерфейс классификатора для задач классификации естественного языка распределяет строки по различным классам.

API ожидает модель TFLite со следующим тензором ввода-вывода:

  • Входной тензор (kTfLiteString)
    • ввод модели принимает строку.
  • Тензор выходных оценок (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
    • выходные оценки для каждого класса, если тип является одним из типов Int, деквантуйте его, если это тип Bool, преобразуйте значения в 0,0 и 1,0 соответственно.
    • может иметь необязательный связанный файл в метаданных для меток, файл должен быть обычным текстовым файлом с одной меткой в ​​строке, количество меток должно соответствовать количеству категорий, выдаваемых моделью. Тензор выходной метки: необязательно (kTfLiteString) — имя выходного класса для каждого класса должно иметь одинаковую длину с оценками. Если этот тензор отсутствует, API использует индексы оценок в качестве имен классов. - будет игнорироваться, если выходной тензор оценок уже имеет связанный файл метки.
  • Дополнительный тензор выходной метки (kTfLiteString/kTfLiteInt32)
    • имя выходного класса для каждого класса должно иметь ту же длину, что и оценки. Если этот тензор отсутствует, API использует индексы оценок в качестве имен классов.
    • будет игнорироваться, если выходной тензор оценок уже имеет связанный файл метки.

По умолчанию API пытается найти тензоры ввода-вывода с конфигурациями по умолчанию в NLClassifier.NLClassifierOptions , причем имя тензора имеет приоритет над индексом тензора. Опцию можно настроить для разных моделей TFLite.

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

сорт НЛклассификатор.NLClassifierOptions Параметры для идентификации входных и выходных тензоров модели.

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

Список < Категория >
классифицировать ( строковый текст)
Выполняет классификацию строкового ввода, возвращает классифицированные Category .
статический NLClassifier
createFromBufferAndOptions (опции ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )
Создает NLClassifier с моделью ByteBuffer и NLClassifier.NLClassifierOptions .
статический NLClassifier
createFromFile (контекстный контекст, String modelPath)
Создает NLClassifier из NLClassifier.NLClassifierOptions по умолчанию.
статический NLClassifier
createFromFile ( Файл модели файла)
Создает NLClassifier из NLClassifier.NLClassifierOptions по умолчанию.
статический NLClassifier
createFromFileAndOptions (опции File modelFile, NLClassifier.NLClassifierOptions )
Создает NLClassifier из NLClassifier.NLClassifierOptions .
статический NLClassifier
createFromFileAndOptions (контекст контекста, String modelPath, параметры NLClassifier.NLClassifierOptions )
Создает NLClassifier из NLClassifier.NLClassifierOptions .

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

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

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

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

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

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

Создает NLClassifier с моделью ByteBuffer и NLClassifier.NLClassifierOptions .

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

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

Создает NLClassifier из NLClassifier.NLClassifierOptions по умолчанию.

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

public static NLClassifier createFromFile ( File modelFile)

Создает NLClassifier из NLClassifier.NLClassifierOptions по умолчанию.

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

public static NLClassifier createFromFileAndOptions (опции File modelFile, NLClassifier.NLClassifierOptions )

Создает NLClassifier из NLClassifier.NLClassifierOptions .

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

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

Создает NLClassifier из NLClassifier.NLClassifierOptions .

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