org.deeplearning4j.spark.impl.graph.scoring

Class CGVaeReconstructionProbWithKeyFunction<K>

java.lang.Object

org.deeplearning4j.spark.impl.common.score.BaseVaeScoreWithKeyFunctionAdapter<K>

org.deeplearning4j.spark.impl.common.score.BaseVaeReconstructionProbWithKeyFunctionAdapter<K>

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

All Implemented Interfaces:

java.io.Serializable, org.datavec.spark.functions.FlatMapFunctionAdapter<java.util.Iterator<scala.Tuple2<K,org.nd4j.linalg.api.ndarray.INDArray>>,scala.Tuple2<K,java.lang.Double>>

public class CGVaeReconstructionProbWithKeyFunction<K>
extends BaseVaeReconstructionProbWithKeyFunctionAdapter<K>

Function to calculate the reconstruction probability for a variational autoencoder, that is the first layer in a ComputationGraph.
Note that scoring is batched for computational efficiency.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.spark.impl.common.score.BaseVaeScoreWithKeyFunctionAdapter

jsonConfig, params

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
VariationalAutoencoder	getVaeLayer()

Methods inherited from class org.deeplearning4j.spark.impl.common.score.BaseVaeReconstructionProbWithKeyFunctionAdapter

computeScore

Methods inherited from class org.deeplearning4j.spark.impl.common.score.BaseVaeScoreWithKeyFunctionAdapter

call

Methods inherited from class java.lang.Object

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

Constructor Detail

CGVaeReconstructionProbWithKeyFunction

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

Parameters:

params - MultiLayerNetwork parameters

jsonConfig - MultiLayerConfiguration, as json

useLogProbability - If true: use log probability. False: use raw probability.

batchSize - Batch size to use when scoring

numSamples - Number of samples to use when calling VariationalAutoencoder.reconstructionLogProbability(INDArray, int)

Method Detail

getVaeLayer

public VariationalAutoencoder getVaeLayer()

Specified by:

getVaeLayer in class BaseVaeScoreWithKeyFunctionAdapter<K>