API Pengklasifikasi untuk tugas NLClassification dengan model Bert, mengkategorikan string ke dalam kelas yang berbeda. API mengharapkan model TFLite berbasis Bert dengan metadata yang terisi.
Metadata harus berisi informasi berikut:
- 1 input_process_unit untuk Tokenizer Bagian Kata/Kalimat.
- 3 masukan tensor dengan nama "ids", "mask" dan "segment_ids".
- 1 tensor keluaran tipe float32[1, 2], dengan file label terlampir secara opsional. Jika file label dilampirkan, file tersebut harus berupa file teks biasa dengan satu label per baris, jumlah label harus sesuai dengan jumlah kategori yang dihasilkan model.
Kelas Bersarang
kelas | BertNLClassifier.BertNLClassifierOptions | Opsi untuk mengonfigurasi BertNLClassifier. |
Metode Publik
Metode Warisan
Metode Publik
Daftar publik < Kategori > klasifikasi ( Teks string )
Melakukan klasifikasi pada input string, mengembalikan Category
yang diklasifikasikan.
Parameter
teks | memasukkan teks ke model. |
---|
Kembali
- Daftar hasil Kategori.
publik statis BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)
Membuat BertNLClassifier
dengan buffer model dan BertNLClassifier.BertNLClassifierOptions
default.
Parameter
modelBuffer | ByteBuffer langsung atau MappedByteBuffer model |
---|
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian Argumen Ilegal | jika buffer model bukan ByteBuffer langsung atau MappedByteBuffer |
---|---|
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |
public static BertNLClassifier createFromBufferAndOptions (Opsi ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier
dengan buffer model dan BertNLClassifier.BertNLClassifierOptions
.
Parameter
modelBuffer | ByteBuffer langsung atau MappedByteBuffer model |
---|---|
pilihan | untuk mengkonfigurasi pengklasifikasi |
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian Argumen Ilegal | jika buffer model bukan ByteBuffer langsung atau MappedByteBuffer |
---|---|
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |
public static BertNLClassifier createFromFile (Konteks konteks, String modelPath)
Membuat BertNLClassifier
dari file model dengan metadata dan BertNLClassifier.BertNLClassifierOptions
default.
Parameter
konteks | konteks Android |
---|---|
modelPath | Jalur menuju model klasifikasi |
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian IO | Jika file model gagal dimuat |
---|---|
Pengecualian Argumen Ilegal | jika suatu argumen tidak valid |
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |
BertNLClassifier statis publik createFromFile ( File modelFile)
Membuat BertNLClassifier
dari objek File
dengan metadata dan BertNLClassifier.BertNLClassifierOptions
default.
Parameter
file model | Model klasifikasi Contoh File |
---|
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian IO | Jika file model gagal dimuat |
---|---|
Pengecualian Argumen Ilegal | jika suatu argumen tidak valid |
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |
public static BertNLClassifier createFromFileAndOptions ( File modelFile, opsi BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier
dari objek File
dengan metadata dan BertNLClassifier.BertNLClassifierOptions
.
Parameter
file model | Model klasifikasi Contoh File |
---|---|
pilihan | untuk mengkonfigurasi pengklasifikasi |
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian IO | Jika file model gagal dimuat |
---|---|
Pengecualian Argumen Ilegal | jika suatu argumen tidak valid |
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |
public static BertNLClassifier createFromFileAndOptions (Konteks konteks, String modelPath, opsi BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier
dari file model dengan metadata dan BertNLClassifier.BertNLClassifierOptions
.
Parameter
konteks | konteks Android. |
---|---|
modelPath | Jalur menuju model klasifikasi |
pilihan | untuk mengkonfigurasi pengklasifikasi |
Kembali
- contoh
BertNLClassifier
Melempar
Pengecualian IO | Jika file model gagal dimuat |
---|---|
Pengecualian Argumen Ilegal | jika suatu argumen tidak valid |
IlegalStateException | jika ada kesalahan internal |
Pengecualian Waktu Proses | jika ada kesalahan yang tidak ditentukan |