BertNLClassifier

BertNLClassifier clase pública

API de clasificador para tareas de clasificación NLC con modelos Bert, categoriza cadenas en diferentes clases. La API espera un modelo TFLite basado en Bert con metadatos poblados.

Los metadatos deben contener la siguiente información:

  • 1 input_process_unit para Wordpiece / Sentencepiece Tokenizer.
  • 3 tensores de entrada con los nombres "ids", "máscara" y "segmento_id".
  • 1 tensor de salida de tipo float32 [1, 2], con un archivo de etiqueta adjunto opcionalmente. Si se adjunta un archivo de etiqueta, 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.

Clases anidadas

clase BertNLClassifier.BertNLClassifierOptions Opciones para configurar BertNLClassifier.

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 BertNLClassifier
createFromBuffer ( ByteBuffer modelBuffer)
Crea BertNLClassifier con un buffer de modelo y por defecto BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opciones)
Crea BertNLClassifier con un tampón de modelo y BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
CreateFromFile (contexto Contexto, Cadena modelPath)
Crea BertNLClassifier de un archivo de modelo con metadatos y por defecto BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
CreateFromFile ( Archivo MODELFILE)
Crea BertNLClassifier desde un File objeto con metadatos y por defecto BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromFileAndOptions ( Archivo MODELFILE, BertNLClassifier.BertNLClassifierOptions opciones)
Crea BertNLClassifier desde un File objeto con metadatos y BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromFileAndOptions (contexto Contexto, Cadena modelPath, BertNLClassifier.BertNLClassifierOptions opciones)
Crea BertNLClassifier de un archivo de modelo con metadatos y BertNLClassifier.BertNLClassifierOptions .

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 Introduzca texto en el modelo.
Devoluciones
  • Una lista de resultados de categorías.

public static BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Crea BertNLClassifier con un buffer de modelo y por defecto BertNLClassifier.BertNLClassifierOptions .

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

public static BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opciones)

Crea BertNLClassifier con un tampón de modelo y BertNLClassifier.BertNLClassifierOptions .

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

public static BertNLClassifier CreateFromFile (contexto Contexto, Cadena modelPath)

Crea BertNLClassifier de un archivo de modelo con metadatos y por defecto BertNLClassifier.BertNLClassifierOptions .

Parámetros
contexto Contexto de Android
modelPath Ruta al modelo de clasificación
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 BertNLClassifier CreateFromFile ( Archivo MODELFILE)

Crea BertNLClassifier desde un File objeto con metadatos y por defecto BertNLClassifier.BertNLClassifierOptions .

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 BertNLClassifier createFromFileAndOptions ( Archivo MODELFILE, BertNLClassifier.BertNLClassifierOptions opciones)

Crea BertNLClassifier desde un File objeto con metadatos y BertNLClassifier.BertNLClassifierOptions .

Parámetros
modelFile El modelo de clasificación File instancia
opciones para configurar el clasificador
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 BertNLClassifier createFromFileAndOptions (contexto, Contexto Cadena modelPath, BertNLClassifier.BertNLClassifierOptions opciones)

Crea BertNLClassifier de un archivo de modelo con metadatos y BertNLClassifier.BertNLClassifierOptions .

Parámetros
contexto Contexto de Android.
modelPath Ruta al modelo de clasificación
opciones para configurar el clasificador
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