MultiDataSetIterator

test

  // 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")
    .addOutputOneHot("labelsReader", 0, numOfClasses)
    .build()