Testing DL4J multiiter

TestMultiIter

  // init DL4J seq readers
  val seqWordsReader = new CSVSequenceRecordReader()
  seqWordsReader.initialize(new FileSplit(new File(wordsFileSavePath)))

  val seqFeatsReader = new CSVSequenceRecordReader()
  seqFeatsReader.initialize(new FileSplit(new File(featsFileSavePath)))

  val seqLabelsReader = new CSVSequenceRecordReader()
  seqLabelsReader.initialize(new FileSplit(new File(labelsFileSavePath)))

  // wrap the readers into multi iterator
  val batchSize = 1
  val numOfClasses = 2
  val multiIter = new RecordReaderMultiDataSetIterator.Builder(batchSize)
    .addSequenceReader("wordsReader", seqWordsReader)
    .addSequenceReader("featsReader", seqFeatsReader)
    .addSequenceReader("labelsReader", seqLabelsReader)
    .addInput("wordsReader")
    .addInput("featsReader")
    .addOutput("labelsReader")
    .build()

  while (multiIter.hasNext()) {
    val ds = multiIter.next()
    println(ds)
  }