Construtor para NLClassifier.NLClassifierOptions
.
Construtores Públicos
Construtor () |
Métodos Públicos
abstrato NLClassifier.NLClassifierOptions | construir () |
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
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 |
---|