NLClassifier

API Classifier pour les tâches de classification en langage naturel, catégorise les chaînes en différentes classes.

L'API attend un modèle TFLite avec le tenseur d'entrée/sortie suivant :

• Tenseur d'entrée (kTfLiteString)
  • entrée du modèle, accepte une chaîne.
• Tenseur du score de sortie (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
  • scores de sortie pour chaque classe, si le type est l'un des types Int, déquantifiez-le, s'il est de type Bool, convertissez les valeurs en 0,0 et 1,0 respectivement.
  • peut avoir un fichier associé facultatif dans les métadonnées pour les étiquettes, le fichier doit être un fichier texte brut avec une étiquette par ligne, le nombre d'étiquettes doit correspondre au nombre de catégories générées par le modèle. Tenseur d'étiquette de sortie : facultatif (kTfLiteString) - le nom de classe de sortie pour chaque classe, doit être de la même longueur avec les scores. Si ce tenseur n'est pas présent, l'API utilise des indices de score comme noms de classe. - sera ignoré si le tenseur de score de sortie a déjà un fichier d'étiquette associé.
• Tenseur d'étiquette de sortie facultatif (kTfLiteString/kTfLiteInt32)
  • Le nom de classe de sortie pour chaque classe doit être de la même longueur que les scores. Si ce tenseur n'est pas présent, l'API utilise des indices de score comme noms de classe.
  • sera ignoré si le tenseur de score de sortie a déjà un fichier labe associé.

Par défaut, l'API essaie de trouver les tenseurs d'entrée/sortie avec les configurations par défaut dans NLClassifier.NLClassifierOptions , avec le nom du tenseur prioritaire sur l'index du tenseur. L'option est configurable pour différents modèles TFLite.