BertQuestionAnswerer

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

Возвращает максимальное количество возможных ответов на заданный вопрос для моделей контроля качества (BERT, Albert и т. д.).

API ожидает модель TFLite на основе Берта с метаданными, содержащими следующую информацию:

  • input_process_units для токенизатора Wordpiece/Sentencepiece — токенизатор Wordpiece можно использовать для модели MobileBert , токенизатор токенизатора Sentencepiece можно использовать для модели Альберта .
  • 3 входных тензора с именами «ids», «mask» и «segment_ids».
  • 2 выходных тензора с именами «end_logits» и «start_logits».

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

сорт BertQuestionAnswerer.BertQuestionAnswererOptions Варианты настройки BertQuestionAnswerer .

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

Список <QaAnswer>
ответ (контекст строки , вопрос строки )
Отвечает на вопрос в зависимости от контекста и возвращает список возможных QaAnswer .
статический BertQuestionAnswerer
createAlbertQuestionAnswererFromFile (контекстный контекст, строковый modelPath, строковое предложениеPieceModelPath)
Создает экземпляр BertQuestionAnswerer с моделью Альберта и файлом модели фрагмента предложения.
статический BertQuestionAnswerer
createBertQuestionAnswererFromFile (контекстный контекст, String modelPath, String vocabPath)
Создает экземпляр BertQuestionAnswerer с моделью Берта и файлом словаря.
статический BertQuestionAnswerer
createFromFile (контекстный контекст, String modelPath)
Создает экземпляр BertQuestionAnswerer на основе BertQuestionAnswerer.BertQuestionAnswererOptions по умолчанию.
статический BertQuestionAnswerer
createFromFile ( Файл модели файла)
Создает экземпляр BertQuestionAnswerer на основе BertQuestionAnswerer.BertQuestionAnswererOptions по умолчанию.
статический BertQuestionAnswerer
статический BertQuestionAnswerer
createFromFileAndOptions (контекст контекста, строка modelPath, параметры BertQuestionAnswerer.BertQuestionAnswererOptions )

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

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

общедоступный список <QaAnswer> ответ (контекст строки , вопрос строки )

Отвечает на вопрос в зависимости от контекста и возвращает список возможных QaAnswer . Может быть пустым, если в данном контексте ответ не найден.

Параметры
контекст контекст, на котором основан вопрос
вопрос вопрос, который нужно задать
Возврат
  • список возможных ответов в QaAnswer

public static BertQuestionAnswerer createAlbertQuestionAnswererFromFile (контекст контекста, строковый modelPath, строковое предложениеPieceModelPath)

Создает экземпляр BertQuestionAnswerer с моделью Альберта и файлом модели фрагмента предложения.

Одна подходящая модель: https://tfhub.dev/tensorflow/lite-model/albert_lite_base/squadv1/1.

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

public static BertQuestionAnswerer createBertQuestionAnswererFromFile (контекстный контекст, String modelPath, String vocabPath)

Создает экземпляр BertQuestionAnswerer с моделью Берта и файлом словаря.

Одна подходящая модель: https://tfhub.dev/tensorflow/lite-model/mobilebert/1/default/1.

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

public static BertQuestionAnswerer createFromFile (контекст контекста, String modelPath)

Создает экземпляр BertQuestionAnswerer на основе BertQuestionAnswerer.BertQuestionAnswererOptions по умолчанию.

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

public static BertQuestionAnswerer createFromFile ( Файл modelFile)

Создает экземпляр BertQuestionAnswerer на основе BertQuestionAnswerer.BertQuestionAnswererOptions по умолчанию.

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

public static BertQuestionAnswerer createFromFileAndOptions (опции File modelFile, BertQuestionAnswerer.BertQuestionAnswererOptions )

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

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

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