- 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 FeedForwardWithKeyFunction<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