API phân loại cho các tác vụ NLClassification với mô hình Bert, phân loại chuỗi thành các lớp khác nhau. API yêu cầu mô hình TFLite dựa trên Bert với siêu dữ liệu được điền sẵn.
Siêu dữ liệu phải chứa các thông tin sau:
- 1 input_process_unit cho Trình mã hóa từ/câu.
- 3 tensor đầu vào có tên "ids", "mask" và "segment_ids".
- 1 tensor đầu ra loại float32[1, 2], với tệp nhãn được đính kèm tùy chọn. Nếu tệp nhãn được đính kèm thì tệp phải là tệp văn bản thuần túy với một nhãn trên mỗi dòng, số lượng nhãn phải khớp với số danh mục mà mô hình xuất ra.
Các lớp lồng nhau
lớp học | BertNLClassifier.BertNLClassifierOptions | Các tùy chọn để cấu hình BertNLClassifier. |
Phương pháp công khai
Phương pháp kế thừa
Phương pháp công khai
Danh sách công khai < Danh mục > phân loại ( Chuỗi văn bản)
Thực hiện phân loại trên đầu vào chuỗi, trả về Danh Category
đã phân loại.
Thông số
chữ | nhập văn bản vào mô hình. |
---|
Trả lại
- Danh sách kết quả Danh mục.
công khai tĩnh BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)
Tạo BertNLClassifier
với bộ đệm mô hình và BertNLClassifier.BertNLClassifierOptions
mặc định.
Thông số
bộ đệm mô hình | ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình |
---|
Trả lại
- một phiên bản
BertNLClassifier
Ném
Ngoại lệ Đối số bất hợp pháp | nếu bộ đệm mô hình không phải là ByteBuffer trực tiếp hoặc MappedByteBuffer |
---|---|
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 |
công khai tĩnh BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions tùy chọn)
Tạo BertNLClassifier
bằng bộ đệm mô hình và BertNLClassifier.BertNLClassifierOptions
.
Thông số
bộ đệm mô hình | ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình |
---|---|
tùy chọn | để cấu hình bộ phân loại |
Trả lại
- một phiên bản
BertNLClassifier
Ném
Ngoại lệ Đối số bất hợp pháp | nếu bộ đệm mô hình không phải là ByteBuffer trực tiếp hoặc MappedByteBuffer |
---|---|
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 BertNLClassifier createFromFile (Ngữ cảnh ngữ cảnh, String modelPath)
Tạo BertNLClassifier
từ một tệp mô hình có siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions
mặc định.
Thông số
bối cảnh | Bối cảnh Android |
---|---|
đường dẫn mô hình | Đường dẫn đến mô hình phân loại |
Trả lại
- một phiên bản
BertNLClassifier
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 |
công khai tĩnh BertNLClassifier createFromFile ( File modelFile)
Tạo BertNLClassifier
từ một đối tượng File
với siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions
mặc định.
Thông số
tập tin mô hình | Mô hình phân loại File instance |
---|
Trả lại
- một phiên bản
BertNLClassifier
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 BertNLClassifier createFromFileAndOptions ( File modelFile, BertNLClassifier.BertNLClassifierOptions tùy chọn)
Tạo BertNLClassifier
từ một đối tượng File
có siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions
.
Thông số
tập tin mô hình | Mô hình phân loại File instance |
---|---|
tùy chọn | để cấu hình bộ phân loại |
Trả lại
- một phiên bản
BertNLClassifier
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 BertNLClassifier createFromFileAndOptions (Ngữ cảnh ngữ cảnh, String modelPath, BertNLClassifier.BertNLClassifierOptions tùy chọn)
Tạo BertNLClassifier
từ một tệp mô hình có siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions
.
Thông số
bối cảnh | Bối cảnh Android. |
---|---|
đường dẫn mô hình | Đường dẫn đến mô hình phân loại |
tùy chọn | để cấu hình bộ phân loại |
Trả lại
- một phiên bản
BertNLClassifier
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 |