org.deeplearning4j.util

Class TestDataSetConsumer

java.lang.Object

org.deeplearning4j.util.TestDataSetConsumer

public class TestDataSetConsumer
extends java.lang.Object

Class that consumes DataSets with specified delays, suitable for testing

Field Summary

Fields

Modifier and Type	Field and Description
protected static org.slf4j.Logger	logger

Constructor Summary

Constructors

Constructor and Description
TestDataSetConsumer(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator, long delay)
TestDataSetConsumer(long delay)

Method Summary

Modifier and Type	Method and Description
long	consumeOnce(org.nd4j.linalg.dataset.DataSet dataSet, boolean consumeWithSleep) This method consumes single DataSet, and spends delay time simulating execution of this dataset
long	consumeWhileHasNext(boolean consumeWithSleep) This method cycles through iterator, whie iterator.hasNext() returns true.
long	getCount()

Methods inherited from class java.lang.Object

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

Field Detail

logger

protected static final org.slf4j.Logger logger

Constructor Detail

TestDataSetConsumer

public TestDataSetConsumer(long delay)

TestDataSetConsumer

public TestDataSetConsumer(@NonNull
                           org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator,
                           long delay)

Method Detail

consumeWhileHasNext

public long consumeWhileHasNext(boolean consumeWithSleep)

This method cycles through iterator, whie iterator.hasNext() returns true. After each cycle execution time is simulated either using Thread.sleep() or empty cycle

Parameters:

consumeWithSleep -

Returns:

consumeOnce

public long consumeOnce(@NonNull
                        org.nd4j.linalg.dataset.DataSet dataSet,
                        boolean consumeWithSleep)

This method consumes single DataSet, and spends delay time simulating execution of this dataset

Parameters:

dataSet -

consumeWithSleep -

Returns:

getCount

public long getCount()