Package org.deeplearning4j.earlystopping.scorecalc

Class VAEReconProbScoreCalculator

java.lang.Object

org.deeplearning4j.earlystopping.scorecalc.base.BaseScoreCalculator<Model>

org.deeplearning4j.earlystopping.scorecalc.VAEReconProbScoreCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<Model>

public class VAEReconProbScoreCalculator
extends BaseScoreCalculator<Model>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field	Description
protected boolean	average
protected boolean	logProb
protected int	reconstructionProbNumSamples

Fields inherited from class org.deeplearning4j.earlystopping.scorecalc.base.BaseScoreCalculator

exampleCount, iterator, mdsIterator, minibatchCount, scoreSum

Constructor Summary

Constructors

Constructor	Description
VAEReconProbScoreCalculator(DataSetIterator iterator, int reconstructionProbNumSamples, boolean logProb)	Constructor for average reconstruction probability
VAEReconProbScoreCalculator(DataSetIterator iterator, int reconstructionProbNumSamples, boolean logProb, boolean average)	Constructor for reconstruction probability

Method Summary

Modifier and Type	Method	Description
protected double	finalScore(double scoreSum, int minibatchCount, int exampleCount)
boolean	minimizeScore()
protected INDArray[]	output(Model network, INDArray[] input, INDArray[] fMask, INDArray[] lMask)
protected INDArray	output(Model network, INDArray input, INDArray fMask, INDArray lMask)
protected void	reset()
protected double	scoreMinibatch(Model network, INDArray[] features, INDArray[] labels, INDArray[] fMask, INDArray[] lMask, INDArray[] output)
protected double	scoreMinibatch(Model net, INDArray features, INDArray labels, INDArray fMask, INDArray lMask, INDArray output)

Methods inherited from class org.deeplearning4j.earlystopping.scorecalc.base.BaseScoreCalculator

arr, calculateScore, get0

Methods inherited from class java.lang.Object

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

Field Detail

reconstructionProbNumSamples

protected final int reconstructionProbNumSamples

logProb

protected final boolean logProb

average

protected final boolean average

Constructor Detail

VAEReconProbScoreCalculator

public VAEReconProbScoreCalculator(DataSetIterator iterator,
                                   int reconstructionProbNumSamples,
                                   boolean logProb)

Constructor for average reconstruction probability

Parameters:

iterator - Iterator

reconstructionProbNumSamples - Number of samples. See VariationalAutoencoder.reconstructionProbability(INDArray, int) for details

logProb - If true: calculate (negative) log probability. False: probability

VAEReconProbScoreCalculator

public VAEReconProbScoreCalculator(DataSetIterator iterator,
                                   int reconstructionProbNumSamples,
                                   boolean logProb,
                                   boolean average)

Constructor for reconstruction probability

Parameters:

iterator - Iterator

reconstructionProbNumSamples - Number of samples. See VariationalAutoencoder.reconstructionProbability(INDArray, int) for details

logProb - If true: calculate (negative) log probability. False: probability

average - If true: return average (log) probability. False: sum of log probability.

Method Detail

reset

protected void reset()

Specified by:

reset in class BaseScoreCalculator<Model>

output

protected INDArray output(Model network,
                          INDArray input,
                          INDArray fMask,
                          INDArray lMask)

Specified by:

output in class BaseScoreCalculator<Model>

output

protected INDArray[] output(Model network,
                            INDArray[] input,
                            INDArray[] fMask,
                            INDArray[] lMask)

Specified by:

output in class BaseScoreCalculator<Model>

scoreMinibatch

protected double scoreMinibatch(Model net,
                                INDArray features,
                                INDArray labels,
                                INDArray fMask,
                                INDArray lMask,
                                INDArray output)

Overrides:

scoreMinibatch in class BaseScoreCalculator<Model>

scoreMinibatch

protected double scoreMinibatch(Model network,
                                INDArray[] features,
                                INDArray[] labels,
                                INDArray[] fMask,
                                INDArray[] lMask,
                                INDArray[] output)

Specified by:

scoreMinibatch in class BaseScoreCalculator<Model>

finalScore

protected double finalScore(double scoreSum,
                            int minibatchCount,
                            int exampleCount)

Specified by:

finalScore in class BaseScoreCalculator<Model>

minimizeScore

public boolean minimizeScore()

Returns:

If true: the score should be minimized. If false: the score should be maximized.