BertNLClassifier

בכיתה ציבורית BertNLClassifier

API לסווג למשימות NLCclassification עם מודלים של Bert, מסווג מחרוזת למחלקות שונות. ה-API מצפה למודל TFLite מבוסס Bert עם מטא נתונים מאוכלסים.

המטא נתונים צריכים להכיל את המידע הבא:

  • 1 input_process_unit עבור Wordpiece/Sentencepiece Tokenizer.
  • 3 טנסור קלט עם שמות "ID", "מסכה" ו-"segment_ids".
  • טנסור פלט אחד מסוג float32[1, 2], עם קובץ תווית מצורף אופציונלי. אם מצורף קובץ תווית, הקובץ צריך להיות קובץ טקסט רגיל עם תווית אחת בכל שורה, מספר התוויות צריך להתאים למספר הקטגוריות שהדגם מוציא.

כיתות מקוננות

מעמד BertNLClassifier.BertNLClassifierOptions אפשרויות להגדרת BertNLClassifier.

שיטות ציבוריות

רשימה < קטגוריה >
לסווג (טקסט מחרוזת )
מבצע סיווג על קלט מחרוזת, מחזיר Category מסווגות.
סטטי BertNLClassifier
createFromBuffer ( ByteBuffer modelBuffer)
יוצר BertNLClassifier עם מאגר מודל וברירת המחדל BertNLClassifier.BertNLClassifierOptions .
סטטי BertNLClassifier
סטטי BertNLClassifier
createFromFile (הקשר הקשר, דגם מחרוזת נתיב)
יוצר את BertNLClassifier מקובץ מודל עם מטא נתונים וברירת המחדל BertNLClassifier.BertNLClassifierOptions .
סטטי BertNLClassifier
createFromFile ( קובץ modelFile)
יוצר BertNLClassifier מאובייקט File עם מטא נתונים וברירת המחדל BertNLClassifier.BertNLClassifierOptions .
סטטי BertNLClassifier
createFromFileAndOptions (אפשרויות קובץ modelFile, BertNLClassifier.BertNLClassifierOptions )
יוצר BertNLClassifier מאובייקט File עם מטא נתונים ו- BertNLClassifier.BertNLClassifierOptions .
סטטי BertNLClassifier
createFromFileAndOptions (הקשר הקשר, דגם מחרוזת נתיב, אפשרויות BertNLClassifier.BertNLClassifierOptions )
יוצר את BertNLClassifier מקובץ מודל עם מטא נתונים ו- BertNLClassifier.BertNLClassifierOptions .

שיטות בירושה

שיטות ציבוריות

רשימה ציבורית < קטגוריה > לסווג ( טקסט מחרוזת )

מבצע סיווג על קלט מחרוזת, מחזיר Category מסווגות.

פרמטרים
טֶקסט הזנת טקסט לדגם.
החזרות
  • רשימה של תוצאות קטגוריה.

ציבורי סטטי BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

יוצר BertNLClassifier עם מאגר מודל וברירת המחדל BertNLClassifier.BertNLClassifierOptions .

פרמטרים
דגםBuffer ByteBuffer ישיר או MappedByteBuffer של הדגם
החזרות
זורק
חריג טיעון לא חוקי אם מאגר המודל אינו ByteBuffer ישיר או MappedByteBuffer
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה

ציבורי סטטי BertNLClassifier createFromBufferAndOptions (אפשרויות ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

יוצר BertNLClassifier עם מאגר מודל ו- BertNLClassifier.BertNLClassifierOptions .

פרמטרים
דגםBuffer ByteBuffer ישיר או MappedByteBuffer של הדגם
אפשרויות כדי להגדיר את המסווג
החזרות
זורק
חריג טיעון לא חוקי אם מאגר המודל אינו ByteBuffer ישיר או MappedByteBuffer
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה

ציבורי סטטי BertNLClassifier createFromFile (הקשר הקשר, דגם מחרוזת נתיב)

יוצר את BertNLClassifier מקובץ מודל עם מטא נתונים וברירת המחדל BertNLClassifier.BertNLClassifierOptions .

פרמטרים
הֶקשֵׁר הקשר אנדרואיד
modelPath הדרך למודל הסיווג
החזרות
זורק
IOException אם קובץ הדגם לא מצליח להיטען
חריג טיעון לא חוקי אם טיעון אינו חוקי
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה

ציבורי סטטי BertNLClassifier createFromFile ( קובץ modelFile)

יוצר BertNLClassifier מאובייקט File עם מטא נתונים וברירת המחדל BertNLClassifier.BertNLClassifierOptions .

פרמטרים
modelFile מודל הסיווג File instance
החזרות
זורק
IOException אם קובץ הדגם לא מצליח להיטען
חריג טיעון לא חוקי אם טיעון אינו חוקי
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה

ציבורי סטטי BertNLClassifier createFromFileAndOptions (אפשרויות דגם FileFile , BertNLClassifier.BertNLClassifierOptions )

יוצר BertNLClassifier מאובייקט File עם מטא נתונים ו- BertNLClassifier.BertNLClassifierOptions .

פרמטרים
modelFile מודל הסיווג File instance
אפשרויות כדי להגדיר את המסווג
החזרות
זורק
IOException אם קובץ הדגם לא מצליח להיטען
חריג טיעון לא חוקי אם טיעון אינו חוקי
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה

ציבורי סטטי BertNLClassifier createFromFileAndOptions (הקשר הקשר, דגם StringPath , BertNLClassifier.BertNLClassifierOptions אפשרויות)

יוצר את BertNLClassifier מקובץ מודל עם מטא נתונים ו- BertNLClassifier.BertNLClassifierOptions .

פרמטרים
הֶקשֵׁר הקשר אנדרואיד.
modelPath הדרך למודל הסיווג
אפשרויות כדי להגדיר את המסווג
החזרות
זורק
IOException אם קובץ הדגם לא מצליח להיטען
חריג טיעון לא חוקי אם טיעון אינו חוקי
חריגה של מדינה בלתי חוקית אם יש שגיאה פנימית
חריגת זמן ריצה אם יש שגיאה אחרת שלא צוינה