NLClassifier

L'API di classificazione per attività di classificazione in linguaggio naturale classifica la stringa in classi diverse.

L'API prevede un modello TFLite con il seguente tensore di input/output:

• Tensore di input (kTfLiteString)
  • input del modello, accetta una stringa.
• Tensore del punteggio di output (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
  • punteggi di output per ciascuna classe, se type è uno dei tipi Int, dequantizzalo, se è di tipo Bool, converti i valori rispettivamente in 0,0 e 1,0.
  • può avere un file associato facoltativo nei metadati per le etichette, il file dovrebbe essere un file di testo semplice con un'etichetta per riga, il numero di etichette dovrebbe corrispondere al numero di categorie restituite dal modello. Tensore etichetta di output: opzionale (kTfLiteString) - nome classe di output per ogni classe, dovrebbe avere la stessa lunghezza dei punteggi. Se questo tensore non è presente, l'API utilizza gli indici di punteggio come nomi di classe. - verrà ignorato se il tensore del punteggio di output ha già un file di etichetta associato.
• Tensore etichetta di output opzionale (kTfLiteString/kTfLiteInt32)
  • il nome della classe di output per ogni classe, dovrebbe avere la stessa lunghezza dei punteggi. Se questo tensore non è presente, l'API utilizza gli indici di punteggio come nomi di classe.
  • verrà ignorato se il tensore del punteggio di output ha già un file etichetta associato.

Per impostazione predefinita, l'API tenta di trovare i tensori di input/output con configurazioni predefinite in NLClassifier.NLClassifierOptions , con il nome del tensore con priorità rispetto all'indice del tensore. L'opzione è configurabile per diversi modelli TFLite.