public class TestDataSetIterator extends Object implements DataSetIterator
Constructor and Description |
---|
TestDataSetIterator(DataSet dataset)
This makes an iterator from the given dataset and batchsize
ONLY for use in tests in nd4j
Initializes with a default batch of 5
|
TestDataSetIterator(DataSet dataset,
int batch) |
Modifier and Type | Method and Description |
---|---|
boolean |
asyncSupported()
Does this DataSetIterator support asynchronous prefetching of multiple DataSet objects?
Most DataSetIterators do, but in some cases it may not make sense to wrap this iterator in an
iterator that does asynchronous prefetching.
|
int |
batch()
Batch size
|
int |
cursor()
The current cursor if applicable
|
List<String> |
getLabels()
Get dataset iterator record reader labels
|
DataSetPreProcessor |
getPreProcessor()
Returns preprocessors, if defined
|
boolean |
hasNext() |
int |
inputColumns()
Input columns for the dataset
|
DataSet |
next() |
DataSet |
next(int num)
Like the standard next method but allows a
customizable number of examples returned
|
int |
numExamples()
Total number of examples in the dataset
|
void |
remove() |
void |
reset()
Resets the iterator back to the beginning
|
boolean |
resetSupported()
Is resetting supported by this DataSetIterator? Many DataSetIterators do support resetting,
but some don't
|
void |
setPreProcessor(DataSetPreProcessor preProcessor)
Set a pre processor
|
int |
totalExamples()
Total examples in the iterator
|
int |
totalOutcomes()
The number of labels for the dataset
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
forEachRemaining
public TestDataSetIterator(DataSet dataset, int batch)
public TestDataSetIterator(DataSet dataset)
dataset
- the dataset to make the iterator frombatch
- the batchsize for the iteratorpublic int totalExamples()
DataSetIterator
totalExamples
in interface DataSetIterator
public int inputColumns()
DataSetIterator
inputColumns
in interface DataSetIterator
public int totalOutcomes()
DataSetIterator
totalOutcomes
in interface DataSetIterator
public boolean resetSupported()
DataSetIterator
resetSupported
in interface DataSetIterator
public boolean asyncSupported()
DataSetIterator
asyncSupported
in interface DataSetIterator
public void reset()
DataSetIterator
reset
in interface DataSetIterator
public int batch()
DataSetIterator
batch
in interface DataSetIterator
public int cursor()
DataSetIterator
cursor
in interface DataSetIterator
public int numExamples()
DataSetIterator
numExamples
in interface DataSetIterator
public void setPreProcessor(DataSetPreProcessor preProcessor)
DataSetIterator
setPreProcessor
in interface DataSetIterator
preProcessor
- a pre processor to setpublic DataSetPreProcessor getPreProcessor()
DataSetIterator
getPreProcessor
in interface DataSetIterator
public List<String> getLabels()
DataSetIterator
getLabels
in interface DataSetIterator
public DataSet next(int num)
DataSetIterator
next
in interface DataSetIterator
num
- the number of examplesCopyright © 2016. All Rights Reserved.