public void train() { DataSetIterator iter = new ProfileIterator(trainingData, confirmationSet, imgDatasets, true); int iEpoch = 0; int nEpochs = 300; while (iEpoch < nEpochs) { System.out.printf("EPOCH %d\n", iEpoch); Evaluation eval = new Evaluation(); while (iter.hasNext()) { DataSet ds = iter.next(); net.train(ds); INDArray predict2 = net.output(ds.getFeatureMatrix()); INDArray labels2 = ds.getLabels(); // eval.evalTimeSeries(labels2, predict2); } iter.reset(); // System.out.println(eval.stats()); iEpoch++; } System.out.println("Fitting : DONE"); }