log.info(model.conf().toYaml());
public static void evaluate(Dl4jMlpClassifier clf, Instances data, double minPerfomance) throws Exception { Instances[] split = TestUtil.splitTrainTest(data); Instances train = split[0]; Instances test = split[1]; clf.buildClassifier(train); Evaluation trainEval = new Evaluation(train); trainEval.evaluateModel(clf, train); Evaluation testEval = new Evaluation(train); testEval.evaluateModel(clf, test); final double testPctCorrect = testEval.pctCorrect(); final double trainPctCorrect = trainEval.pctCorrect(); log.info("Train: {}, Test: {}", trainPctCorrect, testPctCorrect); boolean success = testPctCorrect > minPerfomance && trainPctCorrect > minPerfomance; log.info("Success: " + success); log.info(clf.getModel().conf().toYaml()); Assert.assertTrue("Performance was < " + minPerfomance + ". TestPctCorrect: " + testPctCorrect +", TrainPctCorrect: " + trainPctCorrect, success); }