BertNLClassifier

BertNLClassifier kelas publik

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

Daftar < Kategori >
mengklasifikasikan ( String teks)
Melakukan klasifikasi pada input string, mengembalikan Category yang diklasifikasikan.
BertNLClassifier statis
buatDariBuffer ( ByteBuffer modelBuffer)
Membuat BertNLClassifier dengan buffer model dan BertNLClassifier.BertNLClassifierOptions default.
BertNLClassifier statis
createFromBufferAndOptions ( opsi ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier dengan buffer model dan BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statis
createFromFile (Konteks konteks, String modelPath)
Membuat BertNLClassifier dari file model dengan metadata dan BertNLClassifier.BertNLClassifierOptions default.
BertNLClassifier statis
buatDariFile ( File modelFile)
Membuat BertNLClassifier dari objek File dengan metadata dan BertNLClassifier.BertNLClassifierOptions default.
BertNLClassifier statis
createFromFileAndOptions ( File modelFile, opsi BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier dari objek File dengan metadata dan BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statis
createFromFileAndOptions (Konteks konteks, String modelPath, opsi BertNLClassifier.BertNLClassifierOptions )
Membuat BertNLClassifier dari file model dengan metadata dan BertNLClassifier.BertNLClassifierOptions .

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