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.
- 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) Cria NLClassifier a partir do padrão NLClassifier.NLClassifierOptions . |
NLClassificador estático | createFromFile ( Arquivo modelFile) Cria NLClassifier a partir do padrão NLClassifier.NLClassifierOptions . |
NLClassificador estático | createFromFileAndOptions (opções de arquivo modelFile, NLClassifier.NLClassifierOptions ) Cria NLClassifier a partir de NLClassifier.NLClassifierOptions . |
NLClassificador estático | createFromFileAndOptions (contexto de contexto, String modelPath, opções NLClassifier.NLClassifierOptions ) Cria NLClassifier a partir de NLClassifier.NLClassifierOptions . |
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 )
Cria NLClassifier
com um modelo ByteBuffer
e 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
- Instância
NLClassifier
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
- uma instância
NLClassifier
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
- uma instância
NLClassifier
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 )
Cria NLClassifier
a partir de NLClassifier.NLClassifierOptions
.
Parâmetros
arquivomodelo | o modelo de classificação Instância File |
---|---|
opções | configurações para o modelo |
Devoluções
- uma instância
NLClassifier
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 )
Cria NLClassifier
a partir de 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
- uma instância
NLClassifier
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 |