BertNLClassifier

classe pública BertNLClassifier

API classificadora para tarefas NLClassification com modelos Bert, categoriza strings 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 peça de palavra/frase.
  • 3 tensores de entrada com nomes "ids", "mask" e "segment_ids".
  • 1 tensor de saída do tipo float32[1, 2], com um arquivo de rótulo opcionalmente anexado. Se um arquivo de rótulo for anexado, o arquivo deverá ser um arquivo de texto simples com um rótulo por linha, o número de rótulos deve corresponder ao número de categorias geradas pelo modelo.

Classes aninhadas

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

Métodos Públicos

Lista <Categoria>
classificar ( string texto)
Executa classificação em uma entrada de string, retorna Category classificadas.
BertNLClassificador estático
createFromBuffer ( ByteBuffer modelBuffer)
Cria BertNLClassifier com um buffer de modelo e BertNLClassifier.BertNLClassifierOptions padrão.
BertNLClassificador estático
createFromBufferAndOptions (opções ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Cria BertNLClassifier com um buffer de modelo e BertNLClassifier.BertNLClassifierOptions .
BertNLClassificador estático
createFromFile (contexto de contexto, String modelPath)
Cria BertNLClassifier a partir de um arquivo de modelo com metadados e BertNLClassifier.BertNLClassifierOptions padrão.
BertNLClassificador estático
createFromFile ( Arquivo modelFile)
Cria BertNLClassifier a partir de um objeto File com metadados e BertNLClassifier.BertNLClassifierOptions padrão.
BertNLClassificador estático
createFromFileAndOptions (opções de arquivo modelFile, BertNLClassifier.BertNLClassifierOptions )
Cria BertNLClassifier a partir de um objeto File com metadados e BertNLClassifier.BertNLClassifierOptions .
BertNLClassificador estático
createFromFileAndOptions (contexto de contexto, String modelPath, opções BertNLClassifier.BertNLClassifierOptions )
Cria BertNLClassifier a partir de um arquivo de modelo com metadados e BertNLClassifier.BertNLClassifierOptions .

Métodos herdados

Métodos Públicos

lista pública <categoria> classificar ( string texto)

Executa classificação em uma entrada de string, retorna Category classificadas.

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

public static BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

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

Parâmetros
modeloBuffer um ByteBuffer direto ou um MappedByteBuffer do modelo
Devoluções
Lança
Exceção de argumento ilegal se o buffer do modelo não for um ByteBuffer direto ou um MappedByteBuffer
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

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

Cria BertNLClassifier com um buffer de modelo e BertNLClassifier.BertNLClassifierOptions .

Parâmetros
modeloBuffer um ByteBuffer direto ou um MappedByteBuffer do modelo
opções para configurar o classificador
Devoluções
Lança
Exceção de argumento ilegal se o buffer do modelo não for um ByteBuffer direto ou um MappedByteBuffer
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

public static BertNLClassifier createFromFile (contexto de contexto, String modelPath)

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

Parâmetros
contexto Contexto Android
caminho do modelo 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 for inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

public static BertNLClassifier createFromFile ( Arquivo modelFile)

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

Parâmetros
arquivomodelo O modelo de classificação Instância File
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento for inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

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

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

Parâmetros
arquivomodelo O modelo de classificação Instância File
opções para configurar o classificador
Devoluções
Lança
IOException Se o arquivo do modelo falhar ao carregar
Exceção de argumento ilegal se um argumento for inválido
IllegalStateException se houver um erro interno
Exceção de tempo de execução se houver um erro não especificado

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

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

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