Executa a classificação no texto.
Esta API espera um modelo TFLite com TFLite Model Metadata (opcional) que contenha os tensores de entrada e o tensor de saída obrigatórios (descritos abaixo) e os itens de rótulo opcionais (mas recomendados) como AssociatedFiles com o tipo TENSOR_AXIS_LABELS por tensor de classificação de saída.
Os metadados são necessários para modelos com tensores de entrada int32 porque contêm a unidade do processo de entrada para o tokenizador do modelo. Nenhum metadado é necessário para modelos com tensores de entrada de string.
- Tensores de entrada
- Três tensores de entrada (
kTfLiteInt32
) do formato[batch_size x bert_max_seq_len]
que representam os IDs de entrada, de máscara e de segmento. Essa assinatura de entrada requer uma unidade de processo do tokenizador Bert nos metadados do modelo. - Ou um tensor de entrada (
kTfLiteInt32
) de forma[batch_size x max_seq_len]
que representa os IDs de entrada. Essa assinatura de entrada requer uma unidade de processo tokenizador de regex nos metadados do modelo. - Ou um tensor de entrada (
kTfLiteString
) que não tenha forma ou tenha a forma[1]
contendo a string de entrada.
- Três tensores de entrada (
- Pelo menos um tensor de saída (
kTfLiteFloat32
/kBool
) com:N
classes e forma[1 x N]
- mapa (s) de rótulo opcional(mas recomendado) como AssociatedFile-s com o tipo
TENSOR_AXIS_LABELS, contendo um rótulo por linha. O primeiro AssociatedFile (se houver) é usado para preencher o campo
class_name
dos resultados. O campodisplay_name
é preenchido a partir de AssociatedFile (se houver) com localidade que corresponde ao campodisplay_names_locale
daTextClassifierOptions
usada no momento da criação ("en" por padrão, ou seja, inglês). Se nenhum deles estiver disponível, apenas o campoindex
dos resultados será preenchido.
Classes aninhadas
classe | TextClassifier.TextClassifierOptions | Opções para configurar um TextClassifier . |
Métodos públicos
TextClassifierResult | |
void |
close()
Fecha e limpa o
TextClassifier . |
estático TextClassifier |
createFromFile(contexto de contexto, modelPath String)
Cria uma instância
TextClassifier com base em um arquivo de modelo e o TextClassifier.TextClassifierOptions padrão. |
estático TextClassifier |
createFromFile(contexto de contexto, File modelFile)
Cria uma instância
TextClassifier com base em um arquivo de modelo e o TextClassifier.TextClassifierOptions padrão. |
estático TextClassifier |
createFromOptions(contexto do contexto, opções TextClassifier.TextClassifierOptions)
Cria uma instância de
TextClassifier de TextClassifier.TextClassifierOptions . |
Métodos herdados
Métodos públicos
public TextClassifierResult classificar (String inputText)
Executa a classificação no texto de entrada.
Parâmetros
inputText | um String para processamento.
|
---|
public Static TextClassifier createFromFile (contexto de contexto, String modelPath)
Cria uma instância TextClassifier
com base em um arquivo de modelo e o TextClassifier.TextClassifierOptions
padrão.
Parâmetros
contexto | um ERROR(/Context) do Android. |
---|---|
modelPath | caminho para o modelo de texto com metadados nos recursos. |
Gera
se houver um erro durante a criação de TextClassifier .
|
público estático TextClassifier createFromFile (contexto de contexto, arquivo modelFile)
Cria uma instância TextClassifier
com base em um arquivo de modelo e o TextClassifier.TextClassifierOptions
padrão.
Parâmetros
contexto | um ERROR(/Context) do Android. |
---|---|
modelFile | a instância File do modelo de texto. |
Gera
IOException | se ocorrer um erro de E/S ao abrir o arquivo do modelo do tflite. |
---|---|
se houver um erro durante a criação de TextClassifier .
|
public Static TextClassifier createFromOptions (contexto de contexto, TextClassifier.TextClassifierOptions opções)
Cria uma instância de TextClassifier
de TextClassifier.TextClassifierOptions
.
Parâmetros
contexto | um ERROR(/Context) do Android. |
---|---|
do modelo. | uma instância TextClassifier.TextClassifierOptions . |
Gera
se houver um erro durante a criação de TextClassifier .
|