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 Description protected int
exampleCount
protected DataSetIterator
iterator
protected MultiDataSetIterator
mdsIterator
protected int
minibatchCount
protected double
scoreSum
-
Constructor Summary
Constructors Modifier Constructor Description protected
BaseScoreCalculator(@NonNull DataSetIterator iterator)
protected
BaseScoreCalculator(@NonNull MultiDataSetIterator iterator)
-
Method Summary
All Methods Static Methods Instance Methods Abstract Methods Concrete Methods Modifier and Type Method Description static INDArray[]
arr(INDArray in)
double
calculateScore(T network)
Calculate the score for the given MultiLayerNetworkprotected 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 interfaceScoreCalculator<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)
-
-