org.deeplearning4j.earlystopping.scorecalc

Class DataSetLossCalculator

java.lang.Object

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

org.deeplearning4j.earlystopping.scorecalc.DataSetLossCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<Model>

public class DataSetLossCalculator
extends BaseScoreCalculator<Model>

Given a DataSetIterator: calculate the total loss for the model on that data set. Can be used for both MultiLayerNetwork and ComputationGraph

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

Constructor and Description
DataSetLossCalculator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator dataSetIterator, boolean average) Calculate the score (loss function value) on a given data set (usually a test set)
DataSetLossCalculator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator dataSetIterator, boolean average) Calculate the score (loss function value) on a given data set (usually a test set)

Method Summary

Modifier and Type	Method and Description
protected double	finalScore(double scoreSum, int minibatchCount, int exampleCount)
protected org.nd4j.linalg.api.ndarray.INDArray[]	output(Model 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(Model network, org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray fMask, org.nd4j.linalg.api.ndarray.INDArray lMask)
protected void	reset()
protected double	scoreMinibatch(Model 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)
String	toString()

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

arr, calculateScore, get0, scoreMinibatch

Methods inherited from class java.lang.Object

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

Constructor Detail

DataSetLossCalculator

public DataSetLossCalculator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator dataSetIterator,
                             boolean average)

Calculate the score (loss function value) on a given data set (usually a test set)

Parameters:

dataSetIterator - Data set to calculate the score for

average - Whether to return the average (sum of loss / N) or just (sum of loss)

DataSetLossCalculator

public DataSetLossCalculator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator dataSetIterator,
                             boolean average)

Calculate the score (loss function value) on a given data set (usually a test set)

Parameters:

dataSetIterator - Data set to calculate the score for

average - Whether to return the average (sum of loss / N) or just (sum of loss)

Method Detail

toString

public String toString()

Overrides:

toString in class Object

reset

protected void reset()

Specified by:

reset in class BaseScoreCalculator<Model>

output

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

output

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

scoreMinibatch

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

finalScore

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

Specified by:

finalScore in class BaseScoreCalculator<Model>