org.deeplearning4j.earlystopping.scorecalc.base

Class BaseMLNScoreCalculator

java.lang.Object

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

org.deeplearning4j.earlystopping.scorecalc.base.BaseMLNScoreCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<MultiLayerNetwork>

public abstract class BaseMLNScoreCalculator
extends BaseScoreCalculator<MultiLayerNetwork>

See Also:

Serialized Form

Field Summary

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

exampleCount, iterator, mdsIterator, minibatchCount, scoreSum

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseMLNScoreCalculator(DataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
protected INDArray[]	output(MultiLayerNetwork network, INDArray[] input, INDArray[] fMask, INDArray[] lMask)
protected INDArray	output(MultiLayerNetwork network, INDArray input, INDArray fMask, INDArray lMask)
protected double	scoreMinibatch(MultiLayerNetwork network, INDArray[] features, INDArray[] labels, INDArray[] fMask, INDArray[] lMask, INDArray[] output)

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

arr, calculateScore, finalScore, get0, reset, scoreMinibatch

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

Constructor Detail

BaseMLNScoreCalculator

protected BaseMLNScoreCalculator(DataSetIterator iterator)

Method Detail

output

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

Specified by:

output in class BaseScoreCalculator<MultiLayerNetwork>

scoreMinibatch

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

Specified by:

scoreMinibatch in class BaseScoreCalculator<MultiLayerNetwork>

output

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

Specified by:

output in class BaseScoreCalculator<MultiLayerNetwork>