BertNLClassifier

lớp công khai BertNLClassifier

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

Danh sách < 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.
Bộ phân loại BertNL tĩnh
createFromBuffer ( ByteBuffer modelBuffer)
Tạo BertNLClassifier với bộ đệm mô hình và BertNLClassifier.BertNLClassifierOptions mặc định.
Bộ phân loại BertNL tĩnh
Bộ phân loại BertNL tĩnh
createFromFile (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.
Bộ phân loại BertNL tĩnh
createFromFile ( File modelFile)
Tạo BertNLClassifier từ một đối tượng File có siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions mặc định.
Bộ phân loại BertNL tĩnh
createFromFileAndOptions ( Tùy chọn tệp modelFile, BertNLClassifier.BertNLClassifierOptions )
Tạo BertNLClassifier từ một đối tượng File có siêu dữ liệu và BertNLClassifier.BertNLClassifierOptions .
Bộ phân loại BertNL tĩnh
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 .

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
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
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
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
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
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
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