org.deeplearning4j.spark.impl.computationgraph

Class IterativeReduceFlatMapCG

java.lang.Object

org.deeplearning4j.spark.impl.computationgraph.IterativeReduceFlatMapCG

All Implemented Interfaces:

Serializable, org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>,scala.Tuple3<org.nd4j.linalg.api.ndarray.INDArray,ComputationGraphUpdater,ScoreReport>>

public class IterativeReduceFlatMapCG
extends Object
implements org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>,scala.Tuple3<org.nd4j.linalg.api.ndarray.INDArray,ComputationGraphUpdater,ScoreReport>>

Iterative reduce for ComputationGraph with flat map using map partitions

See Also:

Serialized Form

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor and Description
IterativeReduceFlatMapCG(String json, org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params, org.apache.spark.broadcast.Broadcast<ComputationGraphUpdater> updater) Pass in json configuration and baseline parameters

Method Summary

Modifier and Type	Method and Description
Iterable<scala.Tuple3<org.nd4j.linalg.api.ndarray.INDArray,ComputationGraphUpdater,ScoreReport>>	call(Iterator<org.nd4j.linalg.dataset.api.MultiDataSet> dataSetIterator)

Methods inherited from class java.lang.Object

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

Field Detail

log

protected static org.slf4j.Logger log

Constructor Detail

IterativeReduceFlatMapCG

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

Pass in json configuration and baseline parameters

Parameters:

json - json configuration for the network

params - the parameters to use for the network

Method Detail

call

public Iterable<scala.Tuple3<org.nd4j.linalg.api.ndarray.INDArray,ComputationGraphUpdater,ScoreReport>> call(Iterator<org.nd4j.linalg.dataset.api.MultiDataSet> dataSetIterator)
                                                                                                      throws Exception

Specified by:

call in interface org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>,scala.Tuple3<org.nd4j.linalg.api.ndarray.INDArray,ComputationGraphUpdater,ScoreReport>>

Throws:

Exception