org.deeplearning4j.spark.canova

Class RDDMiniBatches.MiniBatchFunction

java.lang.Object

org.deeplearning4j.spark.canova.RDDMiniBatches.MiniBatchFunction

All Implemented Interfaces:

Serializable, org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.DataSet>,org.nd4j.linalg.dataset.DataSet>

Enclosing class:

RDDMiniBatches

public static class RDDMiniBatches.MiniBatchFunction
extends Object
implements org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.DataSet>,org.nd4j.linalg.dataset.DataSet>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MiniBatchFunction(int batchSize)

Method Summary

Modifier and Type	Method and Description
Iterable<org.nd4j.linalg.dataset.DataSet>	call(Iterator<org.nd4j.linalg.dataset.DataSet> dataSetIterator)

Methods inherited from class java.lang.Object

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

Constructor Detail

MiniBatchFunction

public MiniBatchFunction(int batchSize)

Method Detail

call

public Iterable<org.nd4j.linalg.dataset.DataSet> call(Iterator<org.nd4j.linalg.dataset.DataSet> dataSetIterator)
                                               throws Exception

Specified by:

call in interface org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.DataSet>,org.nd4j.linalg.dataset.DataSet>

Throws:

Exception