Probability of output in nntest

rasmusbergpalm / DeepLearnToolbox

Matlab/Octave toolbox for deep learning. Includes Deep Belief Nets, Stacked Autoencoders, Convolutional Neural Nets, Convolutional Autoencoders and vanilla Neural Nets. Each method has examples to get you started.

BSD 2-Clause "Simplified" License

3.78k stars 2.28k forks source link

Probability of output in nntest #152

Open haoqi opened 9 years ago

haoqi commented 9 years ago

For the classification in nntest after training, this toolbox can not return the probability of the classification results, right??

tambetm commented 9 years ago

Look into nnpredict(), basically after nnff() the nn.a{end} is the probabilities (softmax layer activations). nnpredict() just picks the class with highest probability.

haoqi commented 9 years ago

That's correct