public static class RecordReaderDataSetIterator.Builder extends Object
Modifier and Type | Field and Description |
---|---|
protected int |
batchSize |
protected WritableConverter |
converter |
protected int |
labelIndex |
protected int |
labelIndexTo |
protected int |
maxNumBatches |
protected int |
numPossibleLabels |
protected DataSetPreProcessor |
preProcessor |
protected RecordReader |
recordReader |
protected boolean |
regression |
Constructor and Description |
---|
Builder(@NonNull RecordReader rr,
int batchSize) |
Modifier and Type | Method and Description |
---|---|
RecordReaderDataSetIterator |
build() |
RecordReaderDataSetIterator.Builder |
classification(int labelIndex,
int numClasses)
Use this for classification
|
RecordReaderDataSetIterator.Builder |
collectMetaData(boolean collectMetaData)
When set to true: metadata for the current examples will be present in the returned DataSet.
|
RecordReaderDataSetIterator.Builder |
maxNumBatches(int maxNumBatches)
Optional argument, usually not used.
|
RecordReaderDataSetIterator.Builder |
preProcessor(DataSetPreProcessor preProcessor)
Optional arg.
|
RecordReaderDataSetIterator.Builder |
regression(int labelIndex)
Use this for single output regression (i.e., 1 output/regression target)
|
RecordReaderDataSetIterator.Builder |
regression(int labelIndexFrom,
int labelIndexTo)
Use this for multiple output regression (1 or more output/regression targets).
|
RecordReaderDataSetIterator.Builder |
writableConverter(WritableConverter converter) |
protected RecordReader recordReader
protected WritableConverter converter
protected int batchSize
protected int maxNumBatches
protected int labelIndex
protected int labelIndexTo
protected int numPossibleLabels
protected boolean regression
protected DataSetPreProcessor preProcessor
public Builder(@NonNull @NonNull RecordReader rr, int batchSize)
rr
- Underlying record reader to source data frombatchSize
- Batch size to usepublic RecordReaderDataSetIterator.Builder writableConverter(WritableConverter converter)
public RecordReaderDataSetIterator.Builder maxNumBatches(int maxNumBatches)
maxNumBatches
- Maximum number of minibatches per epoch / resetpublic RecordReaderDataSetIterator.Builder regression(int labelIndex)
labelIndex
- Column index that contains the regression target (indexes start at 0)public RecordReaderDataSetIterator.Builder regression(int labelIndexFrom, int labelIndexTo)
labelIndexFrom
- Column index of the first regression target (indexes start at 0)labelIndexTo
- Column index of the last regression target (inclusive)public RecordReaderDataSetIterator.Builder classification(int labelIndex, int numClasses)
labelIndex
- Index that contains the label index. Column (indexes start from 0) be an integer value,
and contain values 0 to numClasses-1numClasses
- Number of label classes (i.e., number of categories/classes in the dataset)public RecordReaderDataSetIterator.Builder preProcessor(DataSetPreProcessor preProcessor)
preProcessor
- Preprocessor to usepublic RecordReaderDataSetIterator.Builder collectMetaData(boolean collectMetaData)
collectMetaData
- Whether metadata should be collected or notpublic RecordReaderDataSetIterator build()
Copyright © 2020. All rights reserved.