org.deeplearning4j.spark.impl.computationgraph

Class ScoreFlatMapFunctionCGMultiDataSet

java.lang.Object

org.deeplearning4j.spark.impl.computationgraph.ScoreFlatMapFunctionCGMultiDataSet

All Implemented Interfaces:

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

public class ScoreFlatMapFunctionCGMultiDataSet
extends Object
implements org.apache.spark.api.java.function.FlatMapFunction<Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>,Double>

Function used to score a MultiDataSet using a given ComputationGraph

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ScoreFlatMapFunctionCGMultiDataSet(String json, org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params)

Method Summary

Modifier and Type	Method and Description
Iterable<Double>	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

Constructor Detail

ScoreFlatMapFunctionCGMultiDataSet

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

Method Detail

call

public Iterable<Double> 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>,Double>

Throws:

Exception