Classifier API สำหรับงาน NLClassification ด้วยโมเดล Bert จัดหมวดหมู่สตริงเป็นคลาสต่างๆ API คาดว่าจะมีโมเดล TFLite ที่ใช้ Bert พร้อมข้อมูลเมตาที่เติมไว้
ข้อมูลเมตาควรมีข้อมูลต่อไปนี้:
- 1 input_process_unit สำหรับ Tokenizer ของ Wordpiece/Sentencepiece
- เทนเซอร์อินพุต 3 ตัวที่มีชื่อ "ids", "mask" และ "segment_ids"
- เทนเซอร์เอาท์พุต 1 ตัวประเภท float32[1, 2] พร้อมด้วยไฟล์ป้ายกำกับที่แนบมาด้วย หากแนบไฟล์ป้ายกำกับ ไฟล์ควรเป็นไฟล์ข้อความธรรมดาที่มีหนึ่งป้ายกำกับต่อบรรทัด จำนวนป้ายกำกับควรตรงกับจำนวนหมวดหมู่ที่โมเดลส่งออก
คลาสที่ซ้อนกัน
ระดับ | BertNLClassifier.BertNLClassifierตัวเลือก | ตัวเลือกในการกำหนดค่า BertNLClassifier |
วิธีการสาธารณะ
วิธีการสืบทอด
วิธีการสาธารณะ
รายการ สาธารณะ < หมวดหมู่ > แยกประเภท (ข้อความ สตริง )
ดำเนินการจัดหมวดหมู่ในอินพุตสตริง ส่งคืน Category
s ที่จัดประเภท
พารามิเตอร์
ข้อความ | ป้อนข้อความให้กับโมเดล |
---|
การส่งคืน
- รายการผลลัพธ์หมวดหมู่
BertNLClassifier แบบคงที่สาธารณะ createFromBuffer ( ByteBuffer modelBuffer)
สร้าง BertNLClassifier
ด้วยบัฟเฟอร์โมเดลและ BertNLClassifier.BertNLClassifierOptions
เริ่มต้น
พารามิเตอร์
modelBuffer | ByteBuffer โดยตรงหรือ MappedByteBuffer ของโมเดล |
---|
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | หากบัฟเฟอร์โมเดลไม่ใช่ ByteBuffer โดยตรงหรือ MappedByteBuffer |
---|---|
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |
BertNLClassifier แบบคงที่สาธารณะ createFromBufferAndOptions (ตัวเลือก ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier
ด้วยบัฟเฟอร์โมเดลและ BertNLClassifier.BertNLClassifierOptions
พารามิเตอร์
modelBuffer | ByteBuffer โดยตรงหรือ MappedByteBuffer ของโมเดล |
---|---|
ตัวเลือก | เพื่อกำหนดค่าตัวแยกประเภท |
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | หากบัฟเฟอร์โมเดลไม่ใช่ ByteBuffer โดยตรงหรือ MappedByteBuffer |
---|---|
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |
BertNLClassifier แบบคงที่สาธารณะ createFromFile (บริบทบริบท, String modelPath)
สร้าง BertNLClassifier
จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions
เริ่มต้น
พารามิเตอร์
บริบท | บริบทของ Android |
---|---|
โมเดลพาธ | เส้นทางสู่โมเดลการจัดหมวดหมู่ |
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
IOข้อยกเว้น | หากไฟล์โมเดลไม่สามารถโหลดได้ |
---|---|
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | ถ้าอาร์กิวเมนต์ไม่ถูกต้อง |
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |
BertNLClassifier แบบคงที่สาธารณะ createFromFile ( ไฟล์ modelFile)
สร้าง BertNLClassifier
จากอ็อบเจ็กต์ File
ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions
เริ่มต้น
พารามิเตอร์
modelFile | ตัวอย่าง File แบบจำลองการจัดหมวดหมู่ |
---|
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
IOข้อยกเว้น | หากไฟล์โมเดลไม่สามารถโหลดได้ |
---|---|
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | ถ้าอาร์กิวเมนต์ไม่ถูกต้อง |
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |
BertNLClassifier แบบคงที่สาธารณะ createFromFileAndOptions (ตัวเลือก ไฟล์ modelFile, BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier
จากอ็อบเจ็กต์ File
ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions
พารามิเตอร์
modelFile | ตัวอย่าง File แบบจำลองการจัดหมวดหมู่ |
---|---|
ตัวเลือก | เพื่อกำหนดค่าตัวแยกประเภท |
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
IOข้อยกเว้น | หากไฟล์โมเดลไม่สามารถโหลดได้ |
---|---|
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | ถ้าอาร์กิวเมนต์ไม่ถูกต้อง |
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |
BertNLClassifier แบบคงที่สาธารณะ createFromFileAndOptions (บริบทบริบท, String modelPath, ตัวเลือก BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier
จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions
พารามิเตอร์
บริบท | บริบทของ Android |
---|---|
โมเดลพาธ | เส้นทางสู่โมเดลการจัดหมวดหมู่ |
ตัวเลือก | เพื่อกำหนดค่าตัวแยกประเภท |
การส่งคืน
- อินสแตนซ์
BertNLClassifier
ขว้าง
IOข้อยกเว้น | หากไฟล์โมเดลไม่สามารถโหลดได้ |
---|---|
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย | ถ้าอาร์กิวเมนต์ไม่ถูกต้อง |
IllegalStateException | หากมีข้อผิดพลาดภายใน |
RuntimeException | หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น |