2017-07-12 11 views
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(); 

答えて

2

使用model.output 。

あなたはそれはあなたがラベル出力のバッチサイズx番号を取り戻す意味出力にソフトマックスを使用しています(http://nd4j.org/tensor

ndarrayから返されます。

+0

したがって、 'INDArray'内の値は各ラベルの確率ですか? – beatngu13

+0

はい、softmaxがどのように働くかを簡潔にしたいのであれば、それはこの質問の対象外です。 Softmaxはよくカバーされているトピックです。 –

関連する問題