public class ConvolutionInstanceIterator extends DefaultInstanceIterator implements ConvolutionalIterator
Constructor and Description |
---|
ConvolutionInstanceIterator() |
Modifier and Type | Method and Description |
---|---|
void |
enforceValidForZooModel(AbstractZooModel tmpZooModel)
The only one-channel zoo model currently implemented is Dl4JLeNet.
|
org.nd4j.linalg.dataset.api.iterator.DataSetIterator |
getDataSetIterator(weka.core.Instances data,
int seed,
int batchSize)
Returns the actual iterator.
|
int |
getHeight() |
int |
getNumChannels() |
java.lang.String[] |
getOptions()
Gets the current settings of the Classifier.
|
int |
getWidth() |
java.lang.String |
globalInfo() |
java.util.Enumeration<weka.core.Option> |
listOptions()
Returns an enumeration describing the available options.
|
void |
setHeight(int height) |
void |
setNumChannels(int numChannels) |
void |
setOptions(java.lang.String[] options)
Parses a given list of options.
|
void |
setWidth(int width) |
validate
getDataSetIterator, getTrainBatchSize, initialize, setTrainBatchSize
@OptionMetadata(displayName="desired width", description="The desired width of the images (default = 28).", commandLineParamName="width", commandLineParamSynopsis="-width <int>", displayOrder=1) public int getWidth()
getWidth
in interface ConvolutionalIterator
public void setWidth(int width)
@OptionMetadata(displayName="desired height", description="The desired height of the images (default = 28).", commandLineParamName="height", commandLineParamSynopsis="-height <int>", displayOrder=2) public int getHeight()
getHeight
in interface ConvolutionalIterator
public void setHeight(int height)
@OptionMetadata(displayName="desired number of channels", description="The desired number of channels (default = 1).", commandLineParamName="numChannels", commandLineParamSynopsis="-numChannels <int>", displayOrder=3) public int getNumChannels()
getNumChannels
in interface ConvolutionalIterator
public void setNumChannels(int numChannels)
public void enforceValidForZooModel(AbstractZooModel tmpZooModel) throws WrongIteratorException
tmpZooModel
- zoo model we need to check againstWrongIteratorException
public org.nd4j.linalg.dataset.api.iterator.DataSetIterator getDataSetIterator(weka.core.Instances data, int seed, int batchSize)
getDataSetIterator
in class DefaultInstanceIterator
data
- the dataset to useseed
- the seed for the random number generatorbatchSize
- the batch size to usepublic java.util.Enumeration<weka.core.Option> listOptions()
listOptions
in interface weka.core.OptionHandler
listOptions
in class DefaultInstanceIterator
public java.lang.String[] getOptions()
getOptions
in interface weka.core.OptionHandler
getOptions
in class DefaultInstanceIterator
public void setOptions(java.lang.String[] options) throws java.lang.Exception
setOptions
in interface weka.core.OptionHandler
setOptions
in class DefaultInstanceIterator
options
- the list of options as an array of stringsjava.lang.Exception
- if an option is not supportedpublic java.lang.String globalInfo()
globalInfo
in class DefaultInstanceIterator