0
私はDeepLearning4Jでちょっと遊んでいますが、ラベルの代わりにどのようにしてスコアを返すことができるのでしょうか。私はlinear classifier tutorialからコードを使用すると仮定し、私はANNは次のように現在の構成が見えます0または1を標識するために与えられた訓練例えば確率を返すしたいのですが:DL4Jクラシファイアを返す戻り値
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();
したがって、 'INDArray'内の値は各ラベルの確率ですか? – beatngu13
はい、softmaxがどのように働くかを簡潔にしたいのであれば、それはこの質問の対象外です。 Softmaxはよくカバーされているトピックです。 –