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
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
- una instancia
BertNLClassifier
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
- una instancia
BertNLClassifier
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
- una instancia
BertNLClassifier
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
- una instancia
BertNLClassifier
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
- una instancia
BertNLClassifier
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
- una instancia
BertNLClassifier
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 |