Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
MultiDataSetIterator iterator;
RecordReaderMultiDataSetIterator.Builder datasetBuilder = new RecordReaderMultiDataSetIterator.Builder(miniBatchSize)
.addReader("encoderInput", encoder_input_reader)
.addReader("decoderInput", decoder_input_Reader)
.addReader("decoderOutput", decoder_output_Reader);
datasetBuilder.addInputOneHot("encoderInput", 0, vocab_size);
datasetBuilder.addInputOneHot("decoderInput", 0, vocab_size);
datasetBuilder.addOutputOneHot("decoderOutput", 0, vocab_size);
datasetBuilder.sequenceAlignmentMode(RecordReaderMultiDataSetIterator.AlignmentMode.ALIGN_END);
iterator = datasetBuilder.build();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.