org.deeplearning4j.earlystopping.scorecalc

Class AutoencoderScoreCalculator

java.lang.Object

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

org.deeplearning4j.earlystopping.scorecalc.AutoencoderScoreCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<Model>

public class AutoencoderScoreCalculator
extends BaseScoreCalculator<Model>

Score function for a MultiLayerNetwork or ComputationGraph with a single AutoEncoder layer. Calculates the specified RegressionEvaluation.Metric on the layer's reconstructions.

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected RegressionEvaluation	evaluation
protected RegressionEvaluation.Metric	metric

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

exampleCount, iterator, mdsIterator, minibatchCount, scoreSum

Constructor Summary

Constructors

Constructor and Description
AutoencoderScoreCalculator(RegressionEvaluation.Metric metric, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
protected double	finalScore(double scoreSum, int minibatchCount, int exampleCount)
protected org.nd4j.linalg.api.ndarray.INDArray[]	output(Model network, org.nd4j.linalg.api.ndarray.INDArray[] input, org.nd4j.linalg.api.ndarray.INDArray[] fMask, org.nd4j.linalg.api.ndarray.INDArray[] lMask)
protected org.nd4j.linalg.api.ndarray.INDArray	output(Model net, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray fMask, org.nd4j.linalg.api.ndarray.INDArray lMask)
protected void	reset()
protected double	scoreMinibatch(Model network, org.nd4j.linalg.api.ndarray.INDArray[] features, org.nd4j.linalg.api.ndarray.INDArray[] labels, org.nd4j.linalg.api.ndarray.INDArray[] fMask, org.nd4j.linalg.api.ndarray.INDArray[] lMask, org.nd4j.linalg.api.ndarray.INDArray[] output)
protected double	scoreMinibatch(Model network, org.nd4j.linalg.api.ndarray.INDArray features, org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray fMask, org.nd4j.linalg.api.ndarray.INDArray lMask, org.nd4j.linalg.api.ndarray.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

metric

protected final RegressionEvaluation.Metric metric

evaluation

protected RegressionEvaluation evaluation

Constructor Detail

AutoencoderScoreCalculator

public AutoencoderScoreCalculator(RegressionEvaluation.Metric metric,
                                  org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Detail

reset

protected void reset()

Specified by:

reset in class BaseScoreCalculator<Model>

output

protected org.nd4j.linalg.api.ndarray.INDArray output(Model net,
                                                      org.nd4j.linalg.api.ndarray.INDArray input,
                                                      org.nd4j.linalg.api.ndarray.INDArray fMask,
                                                      org.nd4j.linalg.api.ndarray.INDArray lMask)

Specified by:

output in class BaseScoreCalculator<Model>

output

protected org.nd4j.linalg.api.ndarray.INDArray[] output(Model network,
                                                        org.nd4j.linalg.api.ndarray.INDArray[] input,
                                                        org.nd4j.linalg.api.ndarray.INDArray[] fMask,
                                                        org.nd4j.linalg.api.ndarray.INDArray[] lMask)

Specified by:

output in class BaseScoreCalculator<Model>

scoreMinibatch

protected double scoreMinibatch(Model network,
                                org.nd4j.linalg.api.ndarray.INDArray features,
                                org.nd4j.linalg.api.ndarray.INDArray labels,
                                org.nd4j.linalg.api.ndarray.INDArray fMask,
                                org.nd4j.linalg.api.ndarray.INDArray lMask,
                                org.nd4j.linalg.api.ndarray.INDArray output)

Overrides:

scoreMinibatch in class BaseScoreCalculator<Model>

scoreMinibatch

protected double scoreMinibatch(Model network,
                                org.nd4j.linalg.api.ndarray.INDArray[] features,
                                org.nd4j.linalg.api.ndarray.INDArray[] labels,
                                org.nd4j.linalg.api.ndarray.INDArray[] fMask,
                                org.nd4j.linalg.api.ndarray.INDArray[] lMask,
                                org.nd4j.linalg.api.ndarray.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>