private void saveEvaluation(boolean train, IEvaluation[] evaluations) throws Exception { String evalPath = FilenameUtils.concat(outputPath, ("evaluation_" + (train ? "train" : "test"))); //Write evaluations to disk for( int i=0; i<evaluations.length; i++ ){ String path = FilenameUtils.concat(evalPath, "evaluation_" + i + ".txt"); FileUtils.writeStringToFile(new File(path), evaluations[i].stats(), Charset.forName("UTF-8")); } }
protected void evalAtIndex(IEvaluation evaluation, INDArray[] labels, INDArray[] predictions, int index) { evaluation.eval(labels[index], predictions[index]); }
evaluation.eval(labels, out[0], labelMask);
evaluation.eval(labels, out[0], labelMask);
evaluation.eval(labels, out, lMask);