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 org.nd4j.linalg.dataset.api.iterator.DataSetIterator	iterator
protected org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator	mdsIterator
protected int	minibatchCount
protected double	scoreSum

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseScoreCalculator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
protected	BaseScoreCalculator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray[]	arr(org.nd4j.linalg.api.ndarray.INDArray in)
double	calculateScore(T network) Calculate the score for the given MultiLayerNetwork
protected abstract double	finalScore(double scoreSum, int minibatchCount, int exampleCount)
static org.nd4j.linalg.api.ndarray.INDArray	get0(org.nd4j.linalg.api.ndarray.INDArray[] in)
protected abstract org.nd4j.linalg.api.ndarray.INDArray[]	output(T network, org.nd4j.linalg.api.ndarray.INDArray[] input, org.nd4j.linalg.api.ndarray.INDArray[] fMask, org.nd4j.linalg.api.ndarray.INDArray[] lMask)
protected abstract org.nd4j.linalg.api.ndarray.INDArray	output(T network, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray fMask, org.nd4j.linalg.api.ndarray.INDArray lMask)
protected abstract void	reset()
protected abstract double	scoreMinibatch(T 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(T 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 java.lang.Object

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

Field Detail

mdsIterator

protected org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator mdsIterator

iterator

protected org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator

scoreSum

protected double scoreSum

minibatchCount

protected int minibatchCount

exampleCount

protected int exampleCount

Constructor Detail

BaseScoreCalculator

protected BaseScoreCalculator(@NonNull
                              org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

BaseScoreCalculator

protected BaseScoreCalculator(@NonNull
                              org.nd4j.linalg.dataset.api.iterator.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 org.nd4j.linalg.api.ndarray.INDArray output(T network,
                                                               org.nd4j.linalg.api.ndarray.INDArray input,
                                                               org.nd4j.linalg.api.ndarray.INDArray fMask,
                                                               org.nd4j.linalg.api.ndarray.INDArray lMask)

output

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

scoreMinibatch

protected double scoreMinibatch(T 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)

scoreMinibatch

protected abstract double scoreMinibatch(T 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)

finalScore

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

arr

public static org.nd4j.linalg.api.ndarray.INDArray[] arr(org.nd4j.linalg.api.ndarray.INDArray in)

get0

public static org.nd4j.linalg.api.ndarray.INDArray get0(org.nd4j.linalg.api.ndarray.INDArray[] in)