BertQuestionAnswerer

lớp công khai BertQuestionNgười trả lời

Trả về các câu trả lời có thể có nhất cho một câu hỏi nhất định cho các mô hình QA (BERT, Albert, v.v.).

API yêu cầu mô hình TFLite dựa trên Bert với siêu dữ liệu chứa thông tin sau:

  • input_process_units cho Trình mã thông báo câu/từ ngữ - Trình mã thông báo từ ngữ có thể được sử dụng cho mô hình MobileBert , Trình mã thông báo câu lệnh có thể được sử dụng cho mô hình Albert .
  • 3 tensor đầu vào có tên "ids", "mask" và "segment_ids".
  • 2 tensor đầu ra có tên "end_logits" và "start_logits".

Các lớp lồng nhau

lớp học BertQuestionAnswerer.BertQuestionAnswererOptions Các tùy chọn để thiết lập BertQuestionAnswerer .

Phương pháp công khai

Danh sách < QaAnswer >
câu trả lời (Bối cảnh chuỗi , Câu hỏi chuỗi )
Trả lời câu hỏi dựa trên ngữ cảnh và trả về danh sách các QaAnswer có thể có.
tĩnh BertCâu hỏiNgười trả lời
createAlbertQuestionAnswererFromFile (Ngữ cảnh, Chuỗi modelPath, Chuỗi câuPieceModelPath)
Tạo một phiên bản BertQuestionAnswerer với mô hình Albert và tệp mô hình đoạn câu.
tĩnh BertCâu hỏiNgười trả lời
createBertQuestionAnswererFromFile (Ngữ cảnh, String modelPath, String vocabPath)
Tạo một phiên bản BertQuestionAnswerer với mô hình Bert và tệp từ vựng.
tĩnh BertCâu hỏiNgười trả lời
createFromFile (Ngữ cảnh, String modelPath)
Tạo một phiên bản BertQuestionAnswerer từ BertQuestionAnswerer.BertQuestionAnswererOptions mặc định.
tĩnh BertCâu hỏiNgười trả lời
createFromFile ( File modelFile)
Tạo một phiên bản BertQuestionAnswerer từ BertQuestionAnswerer.BertQuestionAnswererOptions mặc định.
tĩnh BertCâu hỏiNgười trả lời
tĩnh BertCâu hỏiNgười trả lời

Phương pháp kế thừa

Phương pháp công khai

Danh sách công khai < QaAnswer > câu trả lời (Bối cảnh chuỗi , Câu hỏi chuỗi )

Trả lời câu hỏi dựa trên ngữ cảnh và trả về danh sách các QaAnswer có thể có. Có thể trống nếu không tìm thấy câu trả lời nào từ ngữ cảnh đã cho.

Thông số
bối cảnh bối cảnh câu hỏi dựa trên
câu hỏi câu hỏi để hỏi
Trả lại
  • danh sách các câu trả lời có thể có trong QaAnswer

tĩnh công khai BertQuestionAnswerer createAlbertQuestionAnswererFromFile (Ngữ cảnh, Chuỗi modelPath, Chuỗi câuPieceModelPath)

Tạo một phiên bản BertQuestionAnswerer với mô hình Albert và tệp mô hình đoạn câu.

Một mô hình phù hợp là: https://tfhub.dev/tensorflow/lite-model/albert_lite_base/squadv1/1

Thông số
bối cảnh bối cảnh android
đường dẫn mô hình đường dẫn tệp đến mô hình Albert. Lưu ý: Không nên nén mô hình
câuPieceModelPath đường dẫn tới file mô hình đoạn câu. Lưu ý: Không nên nén mô hình
Trả lại
Ném
IOException Nếu tệp mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

public static BertQuestionAnswerer createBertQuestionAnswererFromFile (Ngữ cảnh ngữ cảnh, String modelPath, String vocabPath)

Tạo một phiên bản BertQuestionAnswerer với mô hình Bert và tệp từ vựng.

Một mô hình phù hợp là: https://tfhub.dev/tensorflow/lite-model/mobilebert/1/default/1

Thông số
bối cảnh bối cảnh android
đường dẫn mô hình đường dẫn tệp đến mô hình Bert. Lưu ý: Không nên nén mô hình
vocabPath đường dẫn tập tin đến tập tin từ vựng. Lưu ý: Không nên nén file
Trả lại
Ném
IOException Nếu tệp mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

public static BertQuestionAnswerer createFromFile (Ngữ cảnh, String modelPath)

Tạo một phiên bản BertQuestionAnswerer từ BertQuestionAnswerer.BertQuestionAnswererOptions mặc định.

Thông số
bối cảnh bối cảnh android
đường dẫn mô hình đường dẫn tệp đến mô hình có siêu dữ liệu. Lưu ý: Không nên nén mô hình
Trả lại
Ném
IOException nếu tập tin mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

tĩnh công khai BertQuestionAnswerer createFromFile ( File modelFile)

Tạo một phiên bản BertQuestionAnswerer từ BertQuestionAnswerer.BertQuestionAnswererOptions mặc định.

Thông số
tập tin mô hình một đối tượng File của mô hình
Trả lại
Ném
IOException nếu tập tin mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

tĩnh công khai BertQuestionAnswerer createFromFileAndOptions ( Tùy chọn tệp modelFile, BertQuestionAnswerer.BertQuestionAnswererOptions )

Thông số
tập tin mô hình một đối tượng File của mô hình
tùy chọn
Trả lại
Ném
IOException nếu tập tin mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

public static BertQuestionAnswerer createFromFileAndOptions (Ngữ cảnh ngữ cảnh, String modelPath, BertQuestionAnswerer.BertQuestionAnswererOptions tùy chọn)

Thông số
bối cảnh bối cảnh android
đường dẫn mô hình đường dẫn tệp đến mô hình có siêu dữ liệu. Lưu ý: Không nên nén mô hình
tùy chọn
Trả lại
Ném
IOException nếu tập tin mô hình không tải được
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác