NLClassifier.NLClassifierOptions.Builder

classe abstrata estática pública NLClassifier.NLClassifierOptions.Builder

Construtores Públicos

Métodos Públicos

abstrato NLClassifier.NLClassifierOptions
abstrato NLClassifier.NLClassifierOptions.Builder
setBaseOptions ( BaseOptions baseOptions)
Define as opções gerais para configurar APIs de tarefas, como aceleradores.
abstrato NLClassifier.NLClassifierOptions.Builder
setInputTensorIndex (int inputTensorIndex)
Defina o índice do tensor de texto de entrada entre todos os tensores de entrada, se o modelo tiver múltiplas entradas.
abstrato NLClassifier.NLClassifierOptions.Builder
setInputTensorName ( String inputTensorName)
Defina o nome do tensor de texto de entrada, se o modelo tiver múltiplas entradas.
abstrato NLClassifier.NLClassifierOptions.Builder
setOutputLabelTensorIndex (int outputLabelTensorIndex)
Defina o índice do tensor de rótulo de saída opcional entre todos os tensores de saída, se o modelo tiver múltiplas saídas.
abstrato NLClassifier.NLClassifierOptions.Builder
setOutputLabelTensorName ( String outputLabelTensorName)
Defina o nome do tensor do rótulo de saída, se o modelo tiver múltiplas saídas.
abstrato NLClassifier.NLClassifierOptions.Builder
setOutputScoreTensorIndex (int outputScoreTensorIndex)
Defina o índice do tensor de pontuação de saída entre todos os tensores de saída, se o modelo tiver múltiplas saídas.
abstrato NLClassifier.NLClassifierOptions.Builder
setOutputScoreTensorName ( String outputScoreTensorName)
Defina o nome do tensor de pontuação de saída, se o modelo tiver múltiplas saídas.

Métodos herdados

Construtores Públicos

Construtor público ()

Métodos Públicos

compilação abstrata pública NLClassifier.NLClassifierOptions ()

público abstrato NLClassifier.NLClassifierOptions.Builder setBaseOptions ( BaseOptions baseOptions)

Define as opções gerais para configurar APIs de tarefas, como aceleradores.

Parâmetros
baseOptions

público abstrato NLClassifier.NLClassifierOptions.Builder setInputTensorIndex (int inputTensorIndex)

Defina o índice do tensor de texto de entrada entre todos os tensores de entrada, se o modelo tiver múltiplas entradas. Somente o tensor de entrada especificado será usado para inferência; outros tensores de entrada serão ignorados. Dafult para 0.

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

Parâmetros
entradaTensorIndex

público abstrato NLClassifier.NLClassifierOptions.Builder setInputTensorName ( String inputTensorName)

Defina o nome do tensor de texto de entrada, se o modelo tiver múltiplas entradas. Somente o tensor de entrada especificado será usado para inferência; outros tensores de entrada serão ignorados. Dafualt para "INPUT" .

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

Parâmetros
inputTensorName

público abstrato NLClassifier.NLClassifierOptions.Builder setOutputLabelTensorIndex (int outputLabelTensorIndex)

Defina o índice do tensor de rótulo de saída opcional entre todos os tensores de saída, se o modelo tiver múltiplas saídas.

Consulte o documento acima outputLabelTensorName para obter mais informações sobre o que é o tensor do rótulo de saída.

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

outputLabelTensorIndex muda para -1, o que significa desabilitar o tensor do rótulo de saída.

Parâmetros
saídaLabelTensorIndex

público abstrato NLClassifier.NLClassifierOptions.Builder setOutputLabelTensorName ( String outputLabelTensorName)

Defina o nome do tensor do rótulo de saída, se o modelo tiver múltiplas saídas. Altere para "OUTPUT_LABEL" .

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

Por padrão, o arquivo de rótulo deve ser compactado com o tensor de pontuação de saída por meio de metadados do modelo. Consulte o MetadataWriter para NLClassifier . NLClassifier lê e analisa rótulos do arquivo de rótulos automaticamente. No entanto, alguns modelos podem gerar um tensor de rótulo específico. Neste caso, NLClassifier lê rótulos do tensor de rótulo de saída.

Parâmetros
saídaLabelTensorName

público abstrato NLClassifier.NLClassifierOptions.Builder setOutputScoreTensorIndex (int outputScoreTensorIndex)

Defina o índice do tensor de pontuação de saída entre todos os tensores de saída, se o modelo tiver múltiplas saídas. Dafult para 0.

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

Parâmetros
saídaScoreTensorIndex

público abstrato NLClassifier.NLClassifierOptions.Builder setOutputScoreTensorName ( String outputScoreTensorName)

Defina o nome do tensor de pontuação de saída, se o modelo tiver múltiplas saídas. Dafualt para "OUTPUT_SCORE" .

Consulte a seção Configurar os tensores de entrada/saída para NLClassifier para obter mais detalhes.

Parâmetros
outputScoreTensorName