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.
- 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 | |
estática NLClassifier | CreateFromFile (contexto Contexto, Cadena modelPath) Crea NLClassifier de defecto NLClassifier.NLClassifierOptions . |
estática NLClassifier | |
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
-
NLClassifier
instancia
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
- un
NLClassifier
instancia
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
- un
NLClassifier
instancia
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
- un
NLClassifier
instancia
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
- un
NLClassifier
instancia
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 |