NLClassifier

API clasificadora para tareas de clasificación de lenguaje natural, clasifica cadenas en diferentes clases.

La API espera un modelo TFLite con el siguiente tensor de entrada/salida:

• Tensor de entrada (kTfLiteString)
  • entrada del modelo, acepta una cadena.
• Tensor de puntuación de salida (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
  • puntuaciones de salida para cada clase, si el tipo es uno de los tipos Int, descuantícelo, si es de tipo Bool, convierta los valores a 0,0 y 1,0 respectivamente.
  • Puede tener un archivo asociado opcional en metadatos para etiquetas, el archivo debe ser un archivo de texto sin formato con una etiqueta por línea, la cantidad de etiquetas debe coincidir con la cantidad de categorías que genera el modelo. Tensor de etiqueta de salida: opcional (kTfLiteString): nombre de clase de salida para cada clase, debe tener la misma longitud que las puntuaciones. Si este tensor no está presente, la API utiliza índices de puntuación como nombres de clase. - se ignorará si el tensor de puntuación de salida ya tiene un archivo de etiqueta asociado.
• Tensor de etiqueta de salida opcional (kTfLiteString/kTfLiteInt32)
  • El nombre de clase de salida para cada clase debe tener la misma longitud que las puntuaciones. Si este tensor no está presente, la API utiliza índices de puntuación como nombres de clase.
  • Se ignorará si el tensor de puntuación de salida ya tiene un archivo de etiqueta asociado.

De forma predeterminada, la API intenta encontrar los tensores de entrada/salida con configuraciones predeterminadas en NLClassifier.NLClassifierOptions , con el nombre del tensor priorizado sobre el índice del tensor. La opción es configurable para diferentes modelos de TFLite.