TextClassifier

clase final pública TextClassifier

Realiza la clasificación del texto.

Esta API espera un modelo TFLite con (opcional) TFLite Model Metadata que contenga los tensores de entrada obligatorios (descritos a continuación), el tensor de salida y los elementos de etiqueta opcionales (pero recomendados) como AssociateFiles con el tipo TENSOR_AXIS_LABELS por tensor de clasificación de salida.

Los metadatos son obligatorios para los modelos con tensores de entrada int32 porque contienen la unidad de proceso de entrada para el Tokenizer del modelo. No se requieren metadatos para los modelos con tensores de entrada de string.

  • Tensores de entrada
    • Tres tensores de entrada (kTfLiteInt32) de forma [batch_size x bert_max_seq_len] que representan los IDs de entrada, los IDs de enmascaramiento y los IDs de segmento. Esta firma de entrada requiere una unidad de proceso de BERT Tokenizer en los metadatos del modelo.
    • O bien, un tensor de entrada (kTfLiteInt32) de forma [batch_size x max_seq_len] que representa los IDs de entrada. Esta firma de entrada requiere una unidad de proceso del Tokenizador de regex en los metadatos del modelo.
    • O bien, un tensor de entrada (kTfLiteString) sin forma o con la forma [1] que contiene la cadena de entrada.
  • Al menos un tensor de salida (kTfLiteFloat32/kBool) con lo siguiente:
    • Clases N y forma [1 x N]
    • Mapas de etiquetas opcionales (pero recomendados) como AssociatedFile-s con el tipo TENSOR_AXIS_LABELS, que contienen una etiqueta por línea. El primer AssociateFile (si existe) se usa para completar el campo class_name de los resultados. El campo display_name se completa desde el AssociatedFile (si existe) cuya configuración regional coincide con el campo display_names_locale del TextClassifierOptions utilizado en el momento de la creación ("en" de forma predeterminada, es decir, inglés). Si ninguna de estas opciones está disponible, solo se completará el campo index de los resultados.

Categorías anidadas

clase TextClassifier.TextClassifierOptions Opciones para configurar un TextClassifier. 

Métodos públicos

TextClassifierResult
classify(String inputText)
Realiza la clasificación del texto de entrada.
void
close()
Cierra y limpia el TextClassifier.
TextClassifier estático
createFromFile(contexto de contexto, String modelPath)
Crea una instancia de TextClassifier a partir de un archivo de modelo y el TextClassifier.TextClassifierOptions predeterminado.
TextClassifier estático
createFromFile(contexto de contexto, Archivo modelFile)
Crea una instancia de TextClassifier a partir de un archivo de modelo y el TextClassifier.TextClassifierOptions predeterminado.
TextClassifier estático
createFromOptions(contexto de contexto, opciones TextClassifier.TextClassifierOptions)
Crea una instancia de TextClassifier a partir de TextClassifier.TextClassifierOptions.

Métodos heredados

Métodos públicos

public TextClassifierResult classify (String inputText)

Realiza la clasificación del texto de entrada.

Parámetros
inputText un String para su procesamiento.

public void close ()

Cierra y limpia el TextClassifier.

publica estática estática TextClassifier createFromFile (contexto de contexto, String modelPath)

Crea una instancia de TextClassifier a partir de un archivo de modelo y el TextClassifier.TextClassifierOptions predeterminado.

Parámetros
Contexto un ERROR(/Context) de Android
modelPath ruta de acceso al modelo de texto con metadatos en los recursos.
Arroja
Si se produce un error durante la creación de TextClassifier.

publica estática estática TextClassifier createFromFile (contexto de contexto, archivo modelFile)

Crea una instancia de TextClassifier a partir de un archivo de modelo y el TextClassifier.TextClassifierOptions predeterminado.

Parámetros
Contexto un ERROR(/Context) de Android
modelFile la instancia del modelo de texto File.
Arroja
IOException si se produce un error de E/S cuando se abre el archivo del modelo de tflite.
Si se produce un error durante la creación de TextClassifier.

estático público TextClassifier createFromOptions (contexto de contexto, opciones de TextClassifier.TextClassifierOptions)

Crea una instancia de TextClassifier a partir de TextClassifier.TextClassifierOptions.

Parámetros
Contexto un ERROR(/Context) de Android
Opciones una instancia de TextClassifier.TextClassifierOptions.
Arroja
Si se produce un error durante la creación de TextClassifier.