NLClassifier

classe pública NLClassifier

API classificadora para tarefas de classificação de linguagem natural, categoriza strings em diferentes classes.

A API espera um modelo TFLite com o seguinte tensor de entrada/saída:

  • Tensor de entrada (kTfLiteString)
    • entrada do modelo, aceita uma string.
  • Tensor de pontuação de saída (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
    • pontuações de saída para cada classe, se o tipo for um dos tipos Int, desquantize-o, se for do tipo Bool, converta os valores para 0,0 e 1,0 respectivamente.
    • pode ter um arquivo opcional associado em metadados para rótulos, o arquivo deve 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 que o modelo gera. Tensor do rótulo de saída: opcional (kTfLiteString) - nome da classe de saída para cada classe, deve ter o mesmo comprimento com pontuações. Se este tensor não estiver presente, a API usa índices de pontuação como nomes de classes. - será ignorado se o tensor de pontuação de saída já tiver um arquivo de rótulo associado.
  • Tensor de rótulo de saída opcional (kTfLiteString/kTfLiteInt32)
    • o nome da classe de saída para cada classe deve ter o mesmo comprimento das pontuações. Se este tensor não estiver presente, a API usa índices de pontuação como nomes de classes.
    • será ignorado se o tensor de pontuação de saída já tiver um arquivo labe associado.

Por padrão, a API tenta encontrar os tensores de entrada/saída com configurações padrão em NLClassifier.NLClassifierOptions , com o nome do tensor priorizado sobre o índice do tensor. A opção é configurável para diferentes modelos de TFLite.

Classes aninhadas

aula NLClassifier.NLClassifierOptions Opções para identificar tensores de entrada e saída do modelo.

Métodos Públicos

Lista <Categoria>
classificar ( string texto)
Executa classificação em uma entrada de string, retorna Category classificadas.
NLClassificador estático
NLClassificador estático
createFromFile (contexto de contexto, String modelPath)
NLClassificador estático
createFromFile ( Arquivo modelFile)
NLClassificador estático
NLClassificador estático

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 inserir texto no modelo
Devoluções
  • uma lista de resultados de categoria

public static NLClassifier createFromBufferAndOptions (opções ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )

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

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

Cria NLClassifier a partir do padrão NLClassifier.NLClassifierOptions .

Parâmetros
contexto Contexto Android
caminho do modelo caminho para o modelo de classificação relativo ao diretório de ativos
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 NLClassifier createFromFile ( Arquivo modelFile)

Cria NLClassifier a partir do padrão NLClassifier.NLClassifierOptions .

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 NLClassifier createFromFileAndOptions (opções de arquivo modelFile, NLClassifier.NLClassifierOptions )

Parâmetros
arquivomodelo o modelo de classificação Instância File
opções configurações para o modelo
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 NLClassifier createFromFileAndOptions (contexto de contexto, String modelPath, opções NLClassifier.NLClassifierOptions )

Parâmetros
contexto Contexto Android
caminho do modelo caminho para o modelo de classificação relativo ao diretório de ativos
opções configurações do modelo.
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