org.deeplearning4j.spark.impl.graph.scoring

Class ScoreExamplesWithKeyFunction<K>

java.lang.Object

org.deeplearning4j.spark.util.BasePairFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.dataset.api.MultiDataSet>>,K,java.lang.Double>

org.deeplearning4j.spark.impl.graph.scoring.ScoreExamplesWithKeyFunction<K>

Type Parameters:

K - Type of key, associated with each example. Used to keep track of which score belongs to which example

All Implemented Interfaces:

java.io.Serializable, org.apache.spark.api.java.function.PairFlatMapFunction<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.dataset.api.MultiDataSet>>,K,java.lang.Double>

public class ScoreExamplesWithKeyFunction<K>
extends BasePairFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.dataset.api.MultiDataSet>>,K,java.lang.Double>

Function to score examples individually, where each example is associated with a particular key
Note that scoring is batched for computational efficiency.
This is the Spark implementation of the ComputationGraph.scoreExamples(MultiDataSet, boolean) method
Note: The MultiDataSet objects passed in must have exactly one example in them (otherwise: can't have a 1:1 association between keys and data sets to score)

See Also:

ScoreExamplesFunction, Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.spark.util.BasePairFlatMapFunctionAdaptee

adapter

Constructor Summary

Constructors

Constructor and Description
ScoreExamplesWithKeyFunction(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params, org.apache.spark.broadcast.Broadcast<java.lang.String> jsonConfig, boolean addRegularizationTerms, int batchSize)

Method Summary

Methods inherited from class org.deeplearning4j.spark.util.BasePairFlatMapFunctionAdaptee

call

Methods inherited from class java.lang.Object

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

Constructor Detail

ScoreExamplesWithKeyFunction

public ScoreExamplesWithKeyFunction(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params,
                                    org.apache.spark.broadcast.Broadcast<java.lang.String> jsonConfig,
                                    boolean addRegularizationTerms,
                                    int batchSize)