dataSetIterator.scala

  val imageRecordReader = new ImageRecordReader(250, 250, 3, labels)
  imageRecordReader.initialize(new FileSplit(rawDir))

  val iterator: DataSetIterator = new RecordReaderDataSetIterator(imageRecordReader, batchSize, 0, labels.length)

  while(iterator.hasNext){
    val nextData: DataSet = iterator.next
    nextData.scale()

    val trainTest = nextData.splitTestAndTrain(0.25d)
    val trainInput = trainTest.getTrain
    testInputBuilder += trainTest.getTest.getFeatureMatrix
    testLabelsBuilder += trainTest.getTest.getLabels
    model.fit(trainInput)
  }