- Use ROC for the model evaluation:
ROC evaluation = new ROC(thresholdSteps);
- Generate output from features in the test data:
DataSet batch = testDataSetIterator.next();
INDArray[] output = model.output(batch.getFeatures());
- Use the ROC evaluation instance to perform the evaluation by calling evalTimeseries():
INDArray actuals = batch.getLabels();
INDArray predictions = output[0]
evaluation.evalTimeSeries(actuals, predictions);
- Display the AUC score (evaluation metrics) by calling calculateAUC():
System.out.println(evaluation.calculateAUC());