public static class RecordReaderMultiDataSetIterator.Builder
extends java.lang.Object
Constructor and Description |
---|
Builder(int batchSize) |
Modifier and Type | Method and Description |
---|---|
RecordReaderMultiDataSetIterator.Builder |
addInput(java.lang.String readerName)
Set as an input, the entire contents (all columns) of the RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addInput(java.lang.String readerName,
int columnFirst,
int columnLast)
Set as an input, a subset of the specified RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addInputOneHot(java.lang.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(java.lang.String readerName)
Set as an output, the entire contents (all columns) of the RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addOutput(java.lang.String readerName,
int columnFirst,
int columnLast)
Add an output, with a subset of the columns from the named RecordReader or SequenceRecordReader
|
RecordReaderMultiDataSetIterator.Builder |
addOutputOneHot(java.lang.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(java.lang.String readerName,
org.datavec.api.records.reader.RecordReader recordReader)
Add a RecordReader for use in .addInput(...) or .addOutput(...)
|
RecordReaderMultiDataSetIterator.Builder |
addSequenceReader(java.lang.String seqReaderName,
org.datavec.api.records.reader.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
|
public Builder(int batchSize)
batchSize
- The batch size for the RecordReaderMultiDataSetIteratorpublic RecordReaderMultiDataSetIterator.Builder addReader(java.lang.String readerName, org.datavec.api.records.reader.RecordReader recordReader)
readerName
- Name of the reader (for later reference)recordReader
- RecordReaderpublic RecordReaderMultiDataSetIterator.Builder addSequenceReader(java.lang.String seqReaderName, org.datavec.api.records.reader.SequenceRecordReader seqRecordReader)
seqReaderName
- Name of the sequence reader (for later reference)seqRecordReader
- SequenceRecordReaderpublic RecordReaderMultiDataSetIterator.Builder sequenceAlignmentMode(RecordReaderMultiDataSetIterator.AlignmentMode alignmentMode)
public RecordReaderMultiDataSetIterator.Builder addInput(java.lang.String readerName)
public RecordReaderMultiDataSetIterator.Builder addInput(java.lang.String readerName, int columnFirst, int columnLast)
readerName
- Name of the readercolumnFirst
- First column index, inclusivecolumnLast
- Last column index, inclusivepublic RecordReaderMultiDataSetIterator.Builder addInputOneHot(java.lang.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(java.lang.String readerName)
public RecordReaderMultiDataSetIterator.Builder addOutput(java.lang.String readerName, int columnFirst, int columnLast)
readerName
- Name of the readercolumnFirst
- First column indexcolumnLast
- Last column index (inclusive)public RecordReaderMultiDataSetIterator.Builder addOutputOneHot(java.lang.String readerName, int column, int numClasses)
readerName
- Name of the RecordReader / SequenceRecordReadercolumn
- index of the columnnumClasses
- Number of classespublic RecordReaderMultiDataSetIterator build()