amten/NeuralNetwork

Test Neural Network

Closed this issue · 1 comments

Hi, I am using your soruce code of neural network in https://github.com/amten/NeuralNetwork
I run the classification example NNClassificationExample.java but I dont know how can I test the neural network after the training phase.

I try with predictedClasses = nn.getPredictedClasses(xCV); and get the next output:

0.999999966203474 1.8422446731342536E-14 2.8438987740204348E-8
1.6397789704005972E-9 0.9999945945378662 1.329060718270638E-7
5.901015821081087E-9 0.9999846853657705 1.087386918868424E-6

How can I classify after train the neural network?

I analyzed your code and saw that
int[] predictedClasses = nn.getPredictedClasses(batchesX.get(batch)); predictedClasses[i] show the predicted classes.