Java 使DL4J分类器返回分数
我在玩一点,我想知道如何让分类器返回分数而不是标签。假设我使用来自的代码,我想让ANN返回给定训练示例的概率,将其标记为0或1。当前配置如下所示:Java 使DL4J分类器返回分数,java,machine-learning,neural-network,deeplearning4j,Java,Machine Learning,Neural Network,Deeplearning4j,我在玩一点,我想知道如何让分类器返回分数而不是标签。假设我使用来自的代码,我想让ANN返回给定训练示例的概率,将其标记为0或1。当前配置如下所示: MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder() .seed(123) .iterations(1) .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIEN
MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
.seed(123)
.iterations(1)
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
.learningRate(0.01)
.updater(Updater.NESTEROVS)
.momentum(0.9)
.list()
.layer(0, new DenseLayer.Builder()
.nIn(2)
.nOut(20)
.weightInit(WeightInit.XAVIER)
.activation(Activation.RELU)
.build())
.layer(1, new OutputLayer.Builder(LossFunction.NEGATIVELOGLIKELIHOOD)
.nIn(20)
.nOut(2)
.weightInit(WeightInit.XAVIER)
.activation(Activation.SOFTMAX)
.build())
.pretrain(false)
.backprop(true)
.build();
使用model.output
你会很快回来的
它在输出上使用softmax,这意味着您将返回批量大小x标签数量的输出。因此,
INDArray
中的值是每个标签的概率?是的,如果您想了解softmax的工作原理,这有点超出了这个问题的范围。Softmax是一个涵盖面很广的主题。