BertCluAnnotator

публичный класс BertCluAnnotator

API для понимания разговорного языка на основе BERT.

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

  • input_process_units для токенизатора Wordpiece.
  • 3 входных тензора с именами «ids», «mask» и «segment_ids».
  • 6 выходных тензоров с именами «domain_task/names», «domain_task/scores», «intent_task/names», «intent_task/scores», «slot_task/names» и «slot_task/scores».

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

сорт BertCluAnnotator.BertCluAnnotatorOptions Варианты настройки BertCluAnnotator .

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

КлюОтклик
аннотировать ( CluRequest cluRequest)
Аннотирует входные высказывания.
статический BertCluAnnotator
createFromBufferAndOptions (опции ByteBuffer modelBuffer, BertCluAnnotator.BertCluAnnotatorOptions )
Создает экземпляр BertCluAnnotator с буфером модели и BertCluAnnotator.BertCluAnnotatorOptions .
статический BertCluAnnotator
createFromFile (контекстный контекст, String modelPath)
Создает экземпляр BertCluAnnotator на основе пути к файлу.

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

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

публичная аннотация CluResponse ( CluRequest cluRequest)

Аннотирует входные высказывания.

Параметры
cluRequest входной диалог, закодированный в CluRequest
Возврат
  • аннотации домена, намерения и слота, закодированные в CluResponse

public static BertCluAnnotator createFromBufferAndOptions (опции ByteBuffer modelBuffer, BertCluAnnotator.BertCluAnnotatorOptions )

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

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

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

Создает экземпляр BertCluAnnotator на основе пути к файлу.

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