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");
}
