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>

Abstract score calculator for MultiLayerNetwonk

Author:

Alex Black

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(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
protected org.nd4j.linalg.api.ndarray.INDArray[]	output(MultiLayerNetwork 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(MultiLayerNetwork network, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray fMask, org.nd4j.linalg.api.ndarray.INDArray lMask)
protected double	scoreMinibatch(MultiLayerNetwork 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, finalScore, get0, reset, scoreMinibatch

Methods inherited from class java.lang.Object

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

Constructor Detail

BaseMLNScoreCalculator

protected BaseMLNScoreCalculator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Detail

output

protected org.nd4j.linalg.api.ndarray.INDArray output(MultiLayerNetwork 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<MultiLayerNetwork>

scoreMinibatch

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

output

protected org.nd4j.linalg.api.ndarray.INDArray[] output(MultiLayerNetwork 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<MultiLayerNetwork>