org.deeplearning4j.earlystopping.scorecalc

Class DataSetLossCalculatorCG

java.lang.Object

org.deeplearning4j.earlystopping.scorecalc.DataSetLossCalculatorCG

All Implemented Interfaces:

Serializable, ScoreCalculator<ComputationGraph>

public class DataSetLossCalculatorCG
extends Object
implements ScoreCalculator<ComputationGraph>

Given a DataSetIterator: calculate the total loss for the model on that data set. Typically used to calculate the loss on a test set.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DataSetLossCalculatorCG(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)
DataSetLossCalculatorCG(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
double	calculateScore(ComputationGraph network) Calculate the score for the given MultiLayerNetwork
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

DataSetLossCalculatorCG

public DataSetLossCalculatorCG(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)

DataSetLossCalculatorCG

public DataSetLossCalculatorCG(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

calculateScore

public double calculateScore(ComputationGraph network)

Description copied from interface: ScoreCalculator

Calculate the score for the given MultiLayerNetwork

Specified by:

calculateScore in interface ScoreCalculator<ComputationGraph>

toString

public String toString()

Overrides:

toString in class Object