org.deeplearning4j.spark.impl.graph.scoring

Class GraphFeedForwardWithKeyFunction<K>

java.lang.Object

org.deeplearning4j.spark.util.BasePairFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.api.ndarray.INDArray[]>>,K,org.nd4j.linalg.api.ndarray.INDArray[]>

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

Type Parameters:

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

All Implemented Interfaces:

java.io.Serializable, org.apache.spark.api.java.function.PairFlatMapFunction<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.api.ndarray.INDArray[]>>,K,org.nd4j.linalg.api.ndarray.INDArray[]>

public class GraphFeedForwardWithKeyFunction<K>
extends BasePairFlatMapFunctionAdaptee<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.api.ndarray.INDArray[]>>,K,org.nd4j.linalg.api.ndarray.INDArray[]>

Function to feed-forward examples, and get the network output (for example, class probabilities). A key value is used to keep track of which output corresponds to which input.

See Also:

Serialized Form

Field Summary

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

adapter

Constructor Summary

Constructors

Constructor and Description
GraphFeedForwardWithKeyFunction(org.apache.spark.broadcast.Broadcast<org.nd4j.linalg.api.ndarray.INDArray> params, org.apache.spark.broadcast.Broadcast<java.lang.String> jsonConfig, 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

GraphFeedForwardWithKeyFunction

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