org.deeplearning4j.earlystopping.scorecalc.base

Class BaseScoreCalculator<T extends Model>

java.lang.Object

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

All Implemented Interfaces:

Serializable, ScoreCalculator<T>

Direct Known Subclasses:

AutoencoderScoreCalculator, BaseMLNScoreCalculator, DataSetLossCalculator, VAEReconErrorScoreCalculator, VAEReconProbScoreCalculator

public abstract class BaseScoreCalculator<T extends Model>
extends Object
implements ScoreCalculator<T>

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int	exampleCount
protected DataSetIterator	iterator
protected MultiDataSetIterator	mdsIterator
protected int	minibatchCount
protected double	scoreSum

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseScoreCalculator(@NonNull DataSetIterator iterator)
protected	BaseScoreCalculator(@NonNull MultiDataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
static INDArray[]	arr(INDArray in)
double	calculateScore(T network) Calculate the score for the given MultiLayerNetwork
protected abstract double	finalScore(double scoreSum, int minibatchCount, int exampleCount)
static INDArray	get0(INDArray[] in)
protected abstract INDArray[]	output(T network, INDArray[] input, INDArray[] fMask, INDArray[] lMask)
protected abstract INDArray	output(T network, INDArray input, INDArray fMask, INDArray lMask)
protected abstract void	reset()
protected abstract double	scoreMinibatch(T network, INDArray[] features, INDArray[] labels, INDArray[] fMask, INDArray[] lMask, INDArray[] output)
protected double	scoreMinibatch(T network, INDArray features, INDArray labels, INDArray fMask, INDArray lMask, INDArray output)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.earlystopping.scorecalc.ScoreCalculator

minimizeScore

Field Detail

mdsIterator

protected MultiDataSetIterator mdsIterator

iterator

protected DataSetIterator iterator

scoreSum

protected double scoreSum

minibatchCount

protected int minibatchCount

exampleCount

protected int exampleCount

Constructor Detail

BaseScoreCalculator

protected BaseScoreCalculator(@NonNull
                              @NonNull DataSetIterator iterator)

BaseScoreCalculator

protected BaseScoreCalculator(@NonNull
                              @NonNull MultiDataSetIterator iterator)

Method Detail

calculateScore

public double calculateScore(T network)

Description copied from interface: ScoreCalculator

Calculate the score for the given MultiLayerNetwork

Specified by:

calculateScore in interface ScoreCalculator<T extends Model>

reset

protected abstract void reset()

output

protected abstract INDArray output(T network,
                                   INDArray input,
                                   INDArray fMask,
                                   INDArray lMask)

output

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

scoreMinibatch

protected double scoreMinibatch(T network,
                                INDArray features,
                                INDArray labels,
                                INDArray fMask,
                                INDArray lMask,
                                INDArray output)

scoreMinibatch

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

finalScore

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

arr

public static INDArray[] arr(INDArray in)

get0

public static INDArray get0(INDArray[] in)