BertNLClassifier

BertNLClassifier classe pública

Classifier API para tarefas de classificação NLC com modelos Bert, categoriza string em diferentes classes. A API espera um modelo TFLite baseado em Bert com metadados preenchidos.

Os metadados devem conter as seguintes informações:

  • 1 input_process_unit para tokenizador de Wordpiece / Frase.
  • 3 tensores de entrada com os nomes "ids", "mask" e "segment_ids".
  • 1 tensor de saída do tipo float32 [1, 2], com um arquivo de etiqueta opcionalmente anexado. Se um arquivo de etiqueta for anexado, o arquivo deve ser um arquivo de texto simples com uma etiqueta por linha, o número de etiquetas deve corresponder ao número de categorias que o modelo produz.

Classes aninhadas

classe BertNLClassifier.BertNLClassifierOptions Opções para configurar BertNLClassifier.

Métodos Públicos

Lista < Categoria >
classificar ( string texto)
Classificação executa em uma string de entrada, retorna classificada Category s.
estática BertNLClassifier
createFromBuffer ( ByteBuffer modelBuffer)
Cria BertNLClassifier com um buffer modelo e padrão BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opções)
Cria BertNLClassifier com um tampão e modelo BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
CreateFromFile (Contexto contexto, corda modelPath)
Cria BertNLClassifier de um arquivo de modelo com metadados e padrão BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
CreateFromFile ( Arquivo modelFile)
Cria BertNLClassifier a partir de um File objeto com metadados e padrão BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromFileAndOptions ( Arquivo modelFile, BertNLClassifier.BertNLClassifierOptions opções)
Cria BertNLClassifier a partir de um File objeto com metadados e BertNLClassifier.BertNLClassifierOptions .
estática BertNLClassifier
createFromFileAndOptions (contexto Contexto, Cordas modelPath, BertNLClassifier.BertNLClassifierOptions opções)
Cria BertNLClassifier de um arquivo de modelo com metadados e BertNLClassifier.BertNLClassifierOptions .

Métodos herdados

Métodos Públicos

pública Lista < Categoria > classificar ( string texto)

Classificação executa em uma string de entrada, retorna classificada Category s.

Parâmetros
texto texto de entrada para o modelo.
Devoluções
  • Uma lista de resultados da categoria.

public static BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Cria BertNLClassifier com um buffer modelo e padrão BertNLClassifier.BertNLClassifierOptions .

Parâmetros
modelBuffer uma direta ByteBuffer ou um MappedByteBuffer do modelo
Devoluções
Lança
Exceção de argumento ilegal Se o tampão de modelo não é um directa ByteBuffer ou um MappedByteBuffer
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma

public static BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opções)

Cria BertNLClassifier com um tampão e modelo BertNLClassifier.BertNLClassifierOptions .

Parâmetros
modelBuffer uma direta ByteBuffer ou um MappedByteBuffer do modelo
opções configurar o classificador
Devoluções
Lança
Exceção de argumento ilegal Se o tampão de modelo não é um directa ByteBuffer ou um MappedByteBuffer
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma

public static BertNLClassifier CreateFromFile (contexto Contexto, Cordas modelPath)

Cria BertNLClassifier de um arquivo de modelo com metadados e padrão BertNLClassifier.BertNLClassifierOptions .

Parâmetros
contexto Contexto Android
modelPath Caminho para o modelo de classificação
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento é inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma

public static BertNLClassifier CreateFromFile ( Arquivo modelFile)

Cria BertNLClassifier a partir de um File objeto com metadados e padrão BertNLClassifier.BertNLClassifierOptions .

Parâmetros
modelFile A classificação modelo File instância
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento é inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma

public static BertNLClassifier createFromFileAndOptions ( Arquivo modelFile, BertNLClassifier.BertNLClassifierOptions opções)

Cria BertNLClassifier a partir de um File objeto com metadados e BertNLClassifier.BertNLClassifierOptions .

Parâmetros
modelFile A classificação modelo File instância
opções configurar o classificador
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento é inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma

public static BertNLClassifier createFromFileAndOptions (contexto contexto, Cordas modelPath, BertNLClassifier.BertNLClassifierOptions opções)

Cria BertNLClassifier de um arquivo de modelo com metadados e BertNLClassifier.BertNLClassifierOptions .

Parâmetros
contexto Contexto Android.
modelPath Caminho para o modelo de classificação
opções configurar o classificador
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento é inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado de outra forma