BertCluAnnotator

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

aula BertCluAnnotator.BertCluAnnotatorOptions Opções para configurar um BertCluAnnotator .

Métodos Públicos

CluResponse
anotar ( CluRequest cluRequest)
Anota as declarações de entrada.
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 a partir de um caminho de arquivo.

Métodos herdados

Métodos Públicos

anotação CluResponse pública ( CluRequest cluRequest)

Anota as declarações 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 do 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 (contexto de contexto, String modelPath)

Cria uma instância BertCluAnnotator a partir de um caminho de arquivo.

Parâmetros
contexto
caminho do modelo caminho do modelo do 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