BertNLClassifier

clase pública BertNLClassifier

API clasificadora para tareas de clasificación NL con modelos Bert, clasifica cadenas en diferentes clases. La API espera un modelo TFLite basado en Bert con metadatos completos.

Los metadatos deben contener la siguiente información:

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

Clases anidadas

clase BertNLClassifier.BertNLClassifierOptions Opciones para configurar BertNLClassifier.

Métodos públicos

Lista < Categoría >
clasificar (texto de cadena )
Realiza la clasificación en una entrada de cadena, devuelve Category clasificadas.
Clasificador BertNL estático
createFromBuffer (modelo ByteBufferBuffer )
Crea BertNLClassifier con un búfer de modelo y BertNLClassifier.BertNLClassifierOptions predeterminado.
Clasificador BertNL estático
createFromBufferAndOptions ( ByteBuffer modelBuffer, opciones BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier con un búfer de modelo y BertNLClassifier.BertNLClassifierOptions .
Clasificador BertNL estático
createFromFile (contexto de contexto, ruta del modelo de cadena )
Crea BertNLClassifier a partir de un archivo de modelo con metadatos y BertNLClassifier.BertNLClassifierOptions predeterminado.
Clasificador BertNL estático
createFromFile ( Archivo modeloArchivo)
Crea BertNLClassifier a partir de un objeto File con metadatos y BertNLClassifier.BertNLClassifierOptions predeterminado.
Clasificador BertNL estático
createFromFileAndOptions ( archivo modelFile, opciones BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier a partir de un objeto File con metadatos y BertNLClassifier.BertNLClassifierOptions .
Clasificador BertNL estático
createFromFileAndOptions (contexto contextual, ruta del modelo de cadena , opciones BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier a partir de un archivo de modelo con metadatos y BertNLClassifier.BertNLClassifierOptions .

Métodos heredados

Métodos públicos

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

Realiza la clasificación en una entrada de cadena, devuelve Category clasificadas.

Parámetros
texto introducir texto en el modelo.
Devoluciones
  • Una lista de resultados de categorías.

BertNLClassifier estático público createFromBuffer ( ByteBuffer modelBuffer)

Crea BertNLClassifier con un búfer de modelo y BertNLClassifier.BertNLClassifierOptions predeterminado.

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

BertNLClassifier estático público createFromBufferAndOptions (opciones ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

Crea BertNLClassifier con un búfer de modelo y BertNLClassifier.BertNLClassifierOptions .

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

BertNLClassifier estático público createFromFile (contexto de contexto, ruta de modelo de cadena )

Crea BertNLClassifier a partir de un archivo de modelo con metadatos y BertNLClassifier.BertNLClassifierOptions predeterminado.

Parámetros
contexto contexto de Android
modeloRuta Camino al modelo de clasificación
Devoluciones
Lanza
IOExcepción Si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
Excepción de estado ilegal si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

BertNLClassifier estático público createFromFile ( archivo modelFile)

Crea BertNLClassifier a partir de un objeto File con metadatos y BertNLClassifier.BertNLClassifierOptions predeterminado.

Parámetros
modeloArchivo El modelo de clasificación Instancia File
Devoluciones
Lanza
IOExcepción Si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
Excepción de estado ilegal si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

BertNLClassifier estático público createFromFileAndOptions (opciones de archivo modelFile, BertNLClassifier.BertNLClassifierOptions )

Crea BertNLClassifier a partir de un objeto File con metadatos y BertNLClassifier.BertNLClassifierOptions .

Parámetros
modeloArchivo El modelo de clasificación Instancia File
opciones para configurar el clasificador
Devoluciones
Lanza
IOExcepción Si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
Excepción de estado ilegal si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado

BertNLClassifier estático público createFromFileAndOptions (contexto de contexto, ruta de modelo de cadena , opciones BertNLClassifier.BertNLClassifierOptions )

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

Parámetros
contexto Contexto de Android.
modeloRuta Camino al modelo de clasificación
opciones para configurar el clasificador
Devoluciones
Lanza
IOExcepción Si el archivo del modelo no se carga
Argumento de excepción ilegal si un argumento no es válido
Excepción de estado ilegal si hay un error interno
Excepción en tiempo de ejecución si hay un error no especificado