Java, AI, Programming, Software Development, Deep Learning.
DataModel data = dataList.get(i); input.putRow(i, Nd4j.create(new double[]{data.getFeature1(), data.getFeature2()})); labels.putRow(i, Nd4j.create(new double[]{data.getLabel()})); } DataSet dataSet = new DataSet(input, labels); DataSetIterator iterator = new ListDataSetIterator<>(dataSet.asList(),