NLClassifier

NLClassifier clase pública

API de clasificador para tareas de clasificación de lenguaje natural, categoriza 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)
    • puntajes de salida para cada clase, si el tipo es uno de los tipos Int, descuantificarlo, si es del 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, el número de etiquetas debe coincidir con el número 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 usa í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)
    • nombre de clase de salida para cada clase, debe tener la misma longitud con las puntuaciones. Si este tensor no está presente, la API usa í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.

Por defecto, los intentos de API para encontrar los tensores de entrada / salida con configuraciones predeterminadas en NLClassifier.NLClassifierOptions , con tensor de prioridad sobre el nombre de tensor índice. La opción es configurable para diferentes modelos de TFLite.

Clases anidadas

clase NLClassifier.NLClassifierOptions Opciones para identificar tensores de entrada y salida del modelo.

Métodos públicos

Lista < Categoría >
clasificar ( cadena de texto)
Realiza la clasificación en una entrada de cadena, regresa clasificado Category s.
estática NLClassifier
createFromBufferAndOptions ( ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions opciones)
Crea NLClassifier con un modelo ByteBuffer y NLClassifier.NLClassifierOptions .
estática NLClassifier
CreateFromFile (contexto Contexto, Cadena modelPath)
Crea NLClassifier de defecto NLClassifier.NLClassifierOptions .
estática NLClassifier
CreateFromFile ( Archivo MODELFILE)
Crea NLClassifier de defecto NLClassifier.NLClassifierOptions .
estática NLClassifier
createFromFileAndOptions ( Archivo MODELFILE, NLClassifier.NLClassifierOptions opciones)
Crea NLClassifier de NLClassifier.NLClassifierOptions .
estática NLClassifier
createFromFileAndOptions (contexto Contexto, Cadena modelPath, NLClassifier.NLClassifierOptions opciones)
Crea NLClassifier de NLClassifier.NLClassifierOptions .

Métodos heredados

Métodos públicos

pública Lista < Categoría > clasificar ( cadena de texto)

Realiza la clasificación en una entrada de cadena, regresa clasificado Category s.

Parámetros
texto introducir texto en el modelo
Devoluciones
  • una lista de resultados de la categoría

public static NLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions opciones)

Crea NLClassifier con un modelo ByteBuffer y NLClassifier.NLClassifierOptions .

Parámetros
modelBuffer una directa ByteBuffer o una MappedByteBuffer del modelo de clasificación
opciones configuraciones para el modelo
Devoluciones
Lanza
IllegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error de otro modo no especificado
Argumento de excepción ilegal si el búfer no es un modelo directo ByteBuffer o una MappedByteBuffer

public static NLClassifier CreateFromFile (contexto Contexto, Cadena modelPath)

Crea NLClassifier de defecto NLClassifier.NLClassifierOptions .

Parámetros
contexto Contexto de Android
modelPath ruta al modelo de clasificación relativa al directorio de activos
Devoluciones
Lanza
IOException si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
IllegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error de otro modo no especificado

public static NLClassifier CreateFromFile ( Archivo MODELFILE)

Crea NLClassifier de defecto NLClassifier.NLClassifierOptions .

Parámetros
modelFile el modelo de clasificación File instancia
Devoluciones
Lanza
IOException si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
IllegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error de otro modo no especificado

public static NLClassifier createFromFileAndOptions ( Archivo MODELFILE, NLClassifier.NLClassifierOptions opciones)

Crea NLClassifier de NLClassifier.NLClassifierOptions .

Parámetros
modelFile el modelo de clasificación File instancia
opciones configuraciones para el modelo
Devoluciones
Lanza
IOException si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
IllegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error de otro modo no especificado

public static NLClassifier createFromFileAndOptions (contexto, Contexto Cadena modelPath, NLClassifier.NLClassifierOptions opciones)

Crea NLClassifier de NLClassifier.NLClassifierOptions .

Parámetros
contexto Contexto de Android
modelPath ruta al modelo de clasificación relativa al directorio de activos
opciones configuraciones para el modelo.
Devoluciones
Lanza
IOException si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
IllegalStateException si hay un error interno
Excepción en tiempo de ejecución si hay un error de otro modo no especificado