classe pública BertCluAnnotator
API para compreensão de linguagem conversacional baseada em BERT.
A API espera um modelo TFLite baseado em Bert com metadados preenchidos. Os metadados devem conter as seguintes informações:
- input_process_units para Wordpiece Tokenizer.
- 3 tensores de entrada com nomes "ids", "mask" e "segment_ids".
- 6 tensores de saída com nomes "domain_task/names", "domain_task/scores", "intent_task/names", "intent_task/scores", "slot_task/names" e "slot_task/scores".
Classes aninhadas
| classe | BertCluAnnotator.BertCluAnnotatorOptions | Opções para configurar um BertCluAnnotator . | |
Métodos públicos
| CluResponse | |
| estático BertCluAnnotator | createFromBufferAndOptions (opções ByteBuffer modelBuffer, BertCluAnnotator.BertCluAnnotatorOptions ) Cria uma instância BertCluAnnotator com um buffer de modelo e BertCluAnnotator.BertCluAnnotatorOptions . |
| estático BertCluAnnotator | createFromFile (Contexto de contexto, String modelPath) Cria uma instância BertCluAnnotator de um caminho de arquivo. |
Métodos Herdados
Métodos públicos
anotação pública CluResponse ( CluRequest cluRequest)
Anota os enunciados de entrada.
Parâmetros
| cluRequest | diálogo de entrada codificado em um CluRequest |
|---|
Devoluções
- anotações de domínio, intenção e slot codificadas em um
CluResponse
public static BertCluAnnotator createFromBufferAndOptions (opções ByteBuffer modelBuffer, BertCluAnnotator.BertCluAnnotatorOptions )
Cria uma instância BertCluAnnotator com um buffer de modelo e BertCluAnnotator.BertCluAnnotatorOptions .
Parâmetros
| modeloBuffer | um ByteBuffer direto ou um MappedByteBuffer do modelo anotador |
|---|---|
| opçõ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 BertCluAnnotator createFromFile (Context context, String modelPath)
Cria uma instância BertCluAnnotator de um caminho de arquivo.
Parâmetros
| contexto | |
|---|---|
| modelPath | caminho do modelo anotador |
Lança
| IOException | se ocorrer um erro de E/S ao carregar o modelo tflite |
|---|---|
| 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 |