Java: Data Science Made Easy

Testing a model

The evaluation of a model is performed using the Evaluation class and the training dataset. An Evaluation instance is created using an argument specifying the number of classes. The test data is fed into the model using the output method. The eval method takes the output of the model and compares it against the test data classes to generate statistics:

Evaluation evaluation = new Evaluation(4); INDArray output = model.output(testData.getFeatureMatrix()); evaluation.eval(testData.getLabels(), output); out.println(evaluation.stats()); 

The output will look similar to the following:

==========================Scores===================================Accuracy: 0.9273Precision: 0.854Recall: 0.8323F1 Score: 0.843

These statistics ...