NLClassifier

NLClassifier kelas publik

API Pengklasifikasi untuk tugas klasifikasi bahasa alami, mengkategorikan string ke dalam kelas yang berbeda.

API mengharapkan model TFLite dengan tensor input/output berikut:

  • Tensor masukan (kTfLiteString)
    • masukan model, menerima string.
  • Tensor skor keluaran (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
    • skor keluaran untuk setiap kelas, jika tipenya adalah salah satu tipe Int, dekuantisasi, jika tipe Bool, ubah nilainya masing-masing menjadi 0,0 dan 1,0.
    • dapat memiliki file terkait opsional dalam metadata untuk label, file tersebut harus berupa file teks biasa dengan satu label per baris, jumlah label harus sesuai dengan jumlah kategori yang dihasilkan model. Tensor label keluaran: opsional (kTfLiteString) - nama kelas keluaran untuk setiap kelas, harus memiliki panjang yang sama dengan skor. Jika tensor ini tidak ada, API akan menggunakan indeks skor sebagai nama kelas. - akan diabaikan jika tensor skor keluaran sudah memiliki file label terkait.
  • Tensor label keluaran opsional (kTfLiteString/kTfLiteInt32)
    • nama kelas keluaran untuk setiap kelas, harus sama panjangnya dengan skor. Jika tensor ini tidak ada, API akan menggunakan indeks skor sebagai nama kelas.
    • akan diabaikan jika tensor skor keluaran sudah memiliki file lab terkait.

Secara default, API mencoba menemukan tensor input/output dengan konfigurasi default di NLClassifier.NLClassifierOptions , dengan nama tensor diprioritaskan daripada indeks tensor. Opsi ini dapat dikonfigurasi untuk model TFLite yang berbeda.

Kelas Bersarang

kelas Opsi NLClassifier.NLClassifier Opsi untuk mengidentifikasi tensor masukan dan keluaran model.

Metode Publik

Daftar < Kategori >
mengklasifikasikan ( String teks)
Melakukan klasifikasi pada input string, mengembalikan Category yang diklasifikasikan.
NLClassifier statis
NLClassifier statis
createFromFile (Konteks konteks, String modelPath)
NLClassifier statis
NLClassifier statis
NLClassifier statis

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

NLClassifier statis publik createFromBufferAndOptions (Opsi ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )

Parameter
modelBuffer ByteBuffer langsung atau MappedByteBuffer dari model klasifikasi
pilihan konfigurasi untuk model
Kembali
Melempar
IlegalStateException jika ada kesalahan internal
Pengecualian Waktu Proses jika ada kesalahan yang tidak ditentukan
Pengecualian Argumen Ilegal jika buffer model bukan ByteBuffer langsung atau MappedByteBuffer

NLClassifier statis publik createFromFile (Konteks konteks, String modelPath)

Parameter
konteks konteks Android
modelPath jalur ke model klasifikasi relatif terhadap direktori aset
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

NLClassifier statis publik createFromFile ( File modelFile)

Parameter
file model contoh File 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

NLClassifier statis publik createFromFileAndOptions ( File modelFile, opsi NLClassifier.NLClassifierOptions )

Parameter
file model contoh File model klasifikasi
pilihan konfigurasi untuk model
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 NLClassifier createFromFileAndOptions (Konteks konteks, String modelPath, opsi NLClassifier.NLClassifierOptions )

Parameter
konteks konteks Android
modelPath jalur ke model klasifikasi relatif terhadap direktori aset
pilihan konfigurasi untuk model.
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