org.deeplearning4j.earlystopping.scorecalc.base

Class BaseIEvaluationScoreCalculator<T extends Model,U extends IEvaluation>

java.lang.Object

org.deeplearning4j.earlystopping.scorecalc.base.BaseIEvaluationScoreCalculator<T,U>

Type Parameters:

T - Type of model

U - Type of evaluation

All Implemented Interfaces:

Serializable, ScoreCalculator<T>

Direct Known Subclasses:

ClassificationScoreCalculator, RegressionScoreCalculator, ROCScoreCalculator

public abstract class BaseIEvaluationScoreCalculator<T extends Model,U extends IEvaluation>
extends Object
implements ScoreCalculator<T>

Base score function based on an IEvaluation instance. Used for both MultiLayerNetwork and ComputationGraph

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.dataset.api.iterator.DataSetIterator	iter
protected org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator	iterator

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseIEvaluationScoreCalculator(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
protected	BaseIEvaluationScoreCalculator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
double	calculateScore(T network) Calculate the score for the given MultiLayerNetwork
protected abstract double	finalScore(U eval)
protected abstract U	newEval()

Methods inherited from class java.lang.Object

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

Field Detail

iterator

protected org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator

iter

protected org.nd4j.linalg.dataset.api.iterator.DataSetIterator iter

Constructor Detail

BaseIEvaluationScoreCalculator

protected BaseIEvaluationScoreCalculator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

BaseIEvaluationScoreCalculator

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

Method Detail

calculateScore

public double calculateScore(T network)

Description copied from interface: ScoreCalculator

Calculate the score for the given MultiLayerNetwork

Specified by:

calculateScore in interface ScoreCalculator<T extends Model>

newEval

protected abstract U newEval()

finalScore

protected abstract double finalScore(U eval)