BertNLClassifier

BertNLClassifier คลาสสาธารณะ

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 เริ่มต้น
BertNLClassifier แบบคงที่
createFromBufferAndOptions (ตัวเลือก ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier ด้วยบัฟเฟอร์โมเดลและ BertNLClassifier.BertNLClassifierOptions
BertNLClassifier แบบคงที่
createFromFile (บริบทบริบท สตริง modelPath)
สร้าง BertNLClassifier จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions เริ่มต้น
BertNLClassifier แบบคงที่
createFromFile ( ไฟล์ โมเดลไฟล์)
สร้าง BertNLClassifier จากอ็อบเจ็กต์ File ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions เริ่มต้น
BertNLClassifier แบบคงที่
createFromFileAndOptions ( ตัวเลือก ไฟล์ modelFile, BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier จากอ็อบเจ็กต์ File ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions
BertNLClassifier แบบคงที่
createFromFileAndOptions (บริบทบริบท ตัวเลือก modelPath สตริง ตัวเลือก BertNLClassifier.BertNLClassifierOptions )
สร้าง BertNLClassifier จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions

วิธีการสืบทอด

วิธีการสาธารณะ

รายการ สาธารณะ < หมวดหมู่ > แยกประเภท (ข้อความ สตริง )

ดำเนินการจัดหมวดหมู่ในอินพุตสตริง ส่งคืน Category s ที่จัดประเภท

พารามิเตอร์
ข้อความ ป้อนข้อความให้กับโมเดล
การส่งคืน
  • รายการผลลัพธ์หมวดหมู่

BertNLClassifier แบบคงที่สาธารณะ createFromBuffer ( ByteBuffer modelBuffer)

สร้าง BertNLClassifier ด้วยบัฟเฟอร์โมเดลและ BertNLClassifier.BertNLClassifierOptions เริ่มต้น

พารามิเตอร์
modelBuffer ByteBuffer โดยตรงหรือ MappedByteBuffer ของโมเดล
การส่งคืน
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย หากบัฟเฟอร์โมเดลไม่ใช่ ByteBuffer โดยตรงหรือ MappedByteBuffer
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น

BertNLClassifier แบบคงที่สาธารณะ createFromBufferAndOptions (ตัวเลือก ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

สร้าง BertNLClassifier ด้วยบัฟเฟอร์โมเดลและ BertNLClassifier.BertNLClassifierOptions

พารามิเตอร์
modelBuffer ByteBuffer โดยตรงหรือ MappedByteBuffer ของโมเดล
ตัวเลือก เพื่อกำหนดค่าตัวแยกประเภท
การส่งคืน
ขว้าง
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย หากบัฟเฟอร์โมเดลไม่ใช่ ByteBuffer โดยตรงหรือ MappedByteBuffer
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น

BertNLClassifier แบบคงที่สาธารณะ createFromFile (บริบทบริบท, String modelPath)

สร้าง BertNLClassifier จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions เริ่มต้น

พารามิเตอร์
บริบท บริบทของ Android
โมเดลพาธ เส้นทางสู่โมเดลการจัดหมวดหมู่
การส่งคืน
ขว้าง
IOข้อยกเว้น หากไฟล์โมเดลไม่สามารถโหลดได้
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย ถ้าอาร์กิวเมนต์ไม่ถูกต้อง
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น

BertNLClassifier แบบคงที่สาธารณะ createFromFile ( ไฟล์ modelFile)

สร้าง BertNLClassifier จากอ็อบเจ็กต์ File ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions เริ่มต้น

พารามิเตอร์
modelFile ตัวอย่าง File แบบจำลองการจัดหมวดหมู่
การส่งคืน
ขว้าง
IOข้อยกเว้น หากไฟล์โมเดลไม่สามารถโหลดได้
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย ถ้าอาร์กิวเมนต์ไม่ถูกต้อง
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น

BertNLClassifier แบบคงที่สาธารณะ createFromFileAndOptions (ตัวเลือก ไฟล์ modelFile, BertNLClassifier.BertNLClassifierOptions )

สร้าง BertNLClassifier จากอ็อบเจ็กต์ File ที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions

พารามิเตอร์
modelFile ตัวอย่าง File แบบจำลองการจัดหมวดหมู่
ตัวเลือก เพื่อกำหนดค่าตัวแยกประเภท
การส่งคืน
ขว้าง
IOข้อยกเว้น หากไฟล์โมเดลไม่สามารถโหลดได้
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย ถ้าอาร์กิวเมนต์ไม่ถูกต้อง
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น

BertNLClassifier แบบคงที่สาธารณะ createFromFileAndOptions (บริบทบริบท, String modelPath, ตัวเลือก BertNLClassifier.BertNLClassifierOptions )

สร้าง BertNLClassifier จากไฟล์โมเดลที่มีข้อมูลเมตาและ BertNLClassifier.BertNLClassifierOptions

พารามิเตอร์
บริบท บริบทของ Android
โมเดลพาธ เส้นทางสู่โมเดลการจัดหมวดหมู่
ตัวเลือก เพื่อกำหนดค่าตัวแยกประเภท
การส่งคืน
ขว้าง
IOข้อยกเว้น หากไฟล์โมเดลไม่สามารถโหลดได้
ข้อยกเว้นอาร์กิวเมนต์ที่ผิดกฎหมาย ถ้าอาร์กิวเมนต์ไม่ถูกต้อง
IllegalStateException หากมีข้อผิดพลาดภายใน
RuntimeException หากมีข้อผิดพลาดที่ไม่ได้ระบุเป็นอย่างอื่น