TextClassifier

Kryen klasifikimin në tekst.

Ky API pret një model TFLite me metadata të modelit TFLite (opsionale) që përmban tensorët e detyrueshëm (të përshkruar më poshtë) në hyrje, tensorin e daljes dhe artikujt opsionalë (por të rekomanduar) të etiketimit si AssociatedFiles me llojin TENSOR_AXIS_LABELS për tensorin e klasifikimit të daljes.

Metadatat kërkohen për modelet me tensorë të hyrjes int32 sepse përmbajnë njësinë e procesit të hyrjes për Tokenizuesin e modelit. Nuk kërkohen meta të dhëna për modelet me tensorë të hyrjes së vargut.

• Tenzoret hyrëse
  • Tre tensorë hyrës ( kTfLiteInt32 ) të formës [batch_size x bert_max_seq_len] që përfaqësojnë ID-të e hyrjes, ID-të e maskave dhe ID-të e segmentit. Ky nënshkrim i hyrjes kërkon një njësi procesi Bert Tokenizer në meta të dhënat e modelit.
  • Ose një tensor hyrës ( kTfLiteInt32 ) me formë [batch_size x max_seq_len] që përfaqëson ID-të e hyrjes. Ky nënshkrim hyrës kërkon një njësi procesi Regex Tokenizer në meta të dhënat e modelit.
  • Ose një tensor hyrës ( kTfLiteString ) që është pa formë ose ka formën [1] që përmban vargun hyrës.
• Të paktën një tensor dalës ( kTfLiteFloat32 / kBool ) me:
  • N klasa dhe forma [1 x N]
  • Hartat e etiketës opsionale (por të rekomanduara) si AssociatedFile-s me llojin TENSOR_AXIS_LABELS, që përmbajnë një etiketë për rresht. I pari i tillë AssociatedFile (nëse ka) përdoret për të mbushur fushën class_name të rezultateve. Fusha display_name plotësohet nga AssociatedFile (nëse ka) vendndodhja e së cilës përputhet me fushën display_names_locale të TextClassifierOptions të përdorura në kohën e krijimit ("en" si parazgjedhje, dmth. anglisht). Nëse asnjëra nga këto nuk është e disponueshme, do të plotësohet vetëm fusha e index të rezultateve.