public static class RecordReaderMultiDataSetIterator.Builder extends Object
| Constructor and Description |
|---|
Builder(int batchSize) |
| Modifier and Type | Method and Description |
|---|---|
RecordReaderMultiDataSetIterator.Builder |
addInput(String readerName)
Set as an input, the entire contents (all columns) of the RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addInput(String readerName,
int columnFirst,
int columnLast)
Set as an input, a subset of the specified RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addInputOneHot(String readerName,
int column,
int numClasses)
Add as an input a single column from the specified RecordReader / SequenceRecordReader
The assumption is that the specified column contains integer values in range 0..numClasses-1;
this integer will be converted to a one-hot representation
|
RecordReaderMultiDataSetIterator.Builder |
addOutput(String readerName)
Set as an output, the entire contents (all columns) of the RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addOutput(String readerName,
int columnFirst,
int columnLast)
Add an output, with a subset of the columns from the named RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addOutputOneHot(String readerName,
int column,
int numClasses)
An an output, where the output is taken from a single column from the specified RecordReader / SequenceRecordReader
The assumption is that the specified column contains integer values in range 0..numClasses-1;
this integer will be converted to a one-hot representation (usually for classification)
|
RecordReaderMultiDataSetIterator.Builder |
addReader(String readerName,
RecordReader recordReader)
Add a RecordReader for use in .addInput(...) or .addOutput(...)
|
RecordReaderMultiDataSetIterator.Builder |
addSequenceReader(String seqReaderName,
SequenceRecordReader seqRecordReader)
Add a SequenceRecordReader for use in .addInput(...) or .addOutput(...)
|
RecordReaderMultiDataSetIterator |
build()
Create the RecordReaderMultiDataSetIterator
|
RecordReaderMultiDataSetIterator.Builder |
sequenceAlignmentMode(RecordReaderMultiDataSetIterator.AlignmentMode alignmentMode)
Set the sequence alignment mode for all sequences
|
RecordReaderMultiDataSetIterator.Builder |
timeSeriesRandomOffset(boolean timeSeriesRandomOffset,
long rngSeed)
For use with timeseries trained with tbptt
In a given minbatch, shorter time series are padded and appropriately masked to be the same length as the longest time series.
|
public Builder(int batchSize)
batchSize - The batch size for the RecordReaderMultiDataSetIteratorpublic RecordReaderMultiDataSetIterator.Builder addReader(String readerName, RecordReader recordReader)
readerName - Name of the reader (for later reference)recordReader - RecordReaderpublic RecordReaderMultiDataSetIterator.Builder addSequenceReader(String seqReaderName, SequenceRecordReader seqRecordReader)
seqReaderName - Name of the sequence reader (for later reference)seqRecordReader - SequenceRecordReaderpublic RecordReaderMultiDataSetIterator.Builder sequenceAlignmentMode(RecordReaderMultiDataSetIterator.AlignmentMode alignmentMode)
public RecordReaderMultiDataSetIterator.Builder addInput(String readerName)
public RecordReaderMultiDataSetIterator.Builder addInput(String readerName, int columnFirst, int columnLast)
readerName - Name of the readercolumnFirst - First column index, inclusivecolumnLast - Last column index, inclusivepublic RecordReaderMultiDataSetIterator.Builder addInputOneHot(String readerName, int column, int numClasses)
readerName - Name of the RecordReader or SequenceRecordReadercolumn - Column that contains the indexnumClasses - Total number of classespublic RecordReaderMultiDataSetIterator.Builder addOutput(String readerName)
public RecordReaderMultiDataSetIterator.Builder addOutput(String readerName, int columnFirst, int columnLast)
readerName - Name of the readercolumnFirst - First column indexcolumnLast - Last column index (inclusive)public RecordReaderMultiDataSetIterator.Builder addOutputOneHot(String readerName, int column, int numClasses)
readerName - Name of the RecordReader / SequenceRecordReadercolumn - index of the columnnumClasses - Number of classespublic RecordReaderMultiDataSetIterator.Builder timeSeriesRandomOffset(boolean timeSeriesRandomOffset, long rngSeed)
timeSeriesRandomOffset, - "true" to randomly offset time series within a minibatchrngSeed - seed for reproducibilitypublic RecordReaderMultiDataSetIterator build()
Copyright © 2018. All rights reserved.