org.deeplearning4j.spark.impl.multilayer

Class IterativeReduce

java.lang.Object

org.deeplearning4j.spark.impl.multilayer.IterativeReduce

All Implemented Interfaces:

Serializable, org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>

public class IterativeReduce
extends Object
implements org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>

Runs iterative reduce on the dataset

Author:

Adam Gibson

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
IterativeReduce(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params, String json) Train and average over mini batches from a dataset

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	call(org.nd4j.linalg.dataset.DataSet dataSet)

Methods inherited from class java.lang.Object

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

Constructor Detail

IterativeReduce

public IterativeReduce(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params,
                       String json)

Train and average over mini batches from a dataset

Parameters:

params - the parameters that were broadcast

json - the configuration for the network

Method Detail

call

public org.nd4j.linalg.api.ndarray.INDArray call(org.nd4j.linalg.dataset.DataSet dataSet)
                                          throws Exception

Specified by:

call in interface org.apache.spark.api.java.function.Function<org.nd4j.linalg.dataset.DataSet,org.nd4j.linalg.api.ndarray.INDArray>

Throws:

Exception