org.deeplearning4j.spark.datavec

Class RDDMiniBatches

java.lang.Object

org.deeplearning4j.spark.datavec.RDDMiniBatches

All Implemented Interfaces:

java.io.Serializable

public class RDDMiniBatches
extends java.lang.Object
implements java.io.Serializable

RDD mini batch partitioning

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	RDDMiniBatches.MiniBatchFunction

Constructor Summary

Constructors

Constructor and Description
RDDMiniBatches(int miniBatches, org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> toSplit)

Method Summary

Modifier and Type	Method and Description
org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet>	miniBatchesJava()

Methods inherited from class java.lang.Object

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

Constructor Detail

RDDMiniBatches

public RDDMiniBatches(int miniBatches,
                      org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> toSplit)

Method Detail

miniBatchesJava

public org.apache.spark.api.java.JavaRDD<org.nd4j.linalg.dataset.DataSet> miniBatchesJava()