org.deeplearning4j.earlystopping.scorecalc

Class DataSetLossCalculator

java.lang.Object

org.deeplearning4j.earlystopping.scorecalc.DataSetLossCalculator

All Implemented Interfaces:

java.io.Serializable, ScoreCalculator<MultiLayerNetwork>

public class DataSetLossCalculator
extends java.lang.Object
implements ScoreCalculator<MultiLayerNetwork>

Given a DataSetIterator: calculate the total loss for the model on that data set. Typically used to calculate the loss on a test set. Note: For early stopping on a ComputationGraph use DataSetLossCalculatorCG

See Also:

Serialized Form

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)

Method Summary

Modifier and Type	Method and Description
double	calculateScore(MultiLayerNetwork network) Calculate the score for the given MultiLayerNetwork
java.lang.String	toString()

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)

Method Detail

calculateScore

public double calculateScore(MultiLayerNetwork network)

Description copied from interface: ScoreCalculator

Calculate the score for the given MultiLayerNetwork

Specified by:

calculateScore in interface ScoreCalculator<MultiLayerNetwork>

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object