публичный класс 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 . |
Публичные методы
КлюОтклик | |
статический 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 | если произошла внутренняя ошибка |
Исключение времени выполнения | если есть неуказанная ошибка |