weka.dl4j.iterators

Class DefaultInstancesIterator

java.lang.Object

weka.dl4j.iterators.AbstractDataSetIterator

weka.dl4j.iterators.DefaultInstancesIterator

All Implemented Interfaces:

java.io.Serializable, OptionHandler

public class DefaultInstancesIterator
extends AbstractDataSetIterator

Converts the given Instances object into a DataSet and then constructs and returns a ShufflingDataSetIterator.

Version:

$Revision: 11711 $

Author:

Christopher Beckham, Eibe Frank

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DefaultInstancesIterator()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.dataset.api.iterator.DataSetIterator	getIterator(Instances data, int seed, int batchSize) Returns the actual iterator.
int	getNumAttributes(Instances data) Returns the number of predictor attributes for this dataset.
int	getTrainBatchSize() Getting the training batch size
void	setTrainBatchSize(int trainBatchSize) Setting the training batch size

Methods inherited from class weka.dl4j.iterators.AbstractDataSetIterator

getIterator, getOptions, listOptions, setOptions

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DefaultInstancesIterator

public DefaultInstancesIterator()

Method Detail

setTrainBatchSize

@OptionMetadata(displayName="size of mini batch",
                description="The mini batch size to use in the iterator (default = 1).",
                commandLineParamName="bs",
                commandLineParamSynopsis="-bs <int>",
                displayOrder=1)
public void setTrainBatchSize(int trainBatchSize)

Description copied from class: AbstractDataSetIterator

Setting the training batch size

Overrides:

setTrainBatchSize in class AbstractDataSetIterator

Parameters:

trainBatchSize - the batch size

getTrainBatchSize

public int getTrainBatchSize()

Description copied from class: AbstractDataSetIterator

Getting the training batch size

Overrides:

getTrainBatchSize in class AbstractDataSetIterator

Returns:

the batch size

getNumAttributes

public int getNumAttributes(Instances data)

Returns the number of predictor attributes for this dataset.

Specified by:

getNumAttributes in class AbstractDataSetIterator

Parameters:

data - the dataset to compute the number of attributes from

Returns:

the number of attributes in the Instances object minus one

getIterator

public org.nd4j.linalg.dataset.api.iterator.DataSetIterator getIterator(Instances data,
                                                                        int seed,
                                                                        int batchSize)

Returns the actual iterator.

Specified by:

getIterator in class AbstractDataSetIterator

Parameters:

data - the dataset to use

seed - the seed for the random number generator

batchSize - the batch size to use

Returns:

the DataSetIterator