org.deeplearning4j.earlystopping.scorecalc

Class RegressionScoreCalculator

java.lang.Object

org.deeplearning4j.earlystopping.scorecalc.base.BaseIEvaluationScoreCalculator<Model,RegressionEvaluation>

org.deeplearning4j.earlystopping.scorecalc.RegressionScoreCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<Model>

public class RegressionScoreCalculator
extends BaseIEvaluationScoreCalculator<Model,RegressionEvaluation>

Calculate the regression score of the network (MultiLayerNetwork or ComputationGraph) on a test set, using the specified regression metric - RegressionEvaluation.Metric

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected RegressionEvaluation.Metric	metric

Fields inherited from class org.deeplearning4j.earlystopping.scorecalc.base.BaseIEvaluationScoreCalculator

iter, iterator

Constructor Summary

Constructors

Constructor and Description
RegressionScoreCalculator(RegressionEvaluation.Metric metric, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
protected double	finalScore(RegressionEvaluation eval)
protected RegressionEvaluation	newEval()

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

calculateScore

Methods inherited from class java.lang.Object

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

Field Detail

metric

protected final RegressionEvaluation.Metric metric

Constructor Detail

RegressionScoreCalculator

public RegressionScoreCalculator(RegressionEvaluation.Metric metric,
                                 org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Method Detail

newEval

protected RegressionEvaluation newEval()

Specified by:

newEval in class BaseIEvaluationScoreCalculator<Model,RegressionEvaluation>

finalScore

protected double finalScore(RegressionEvaluation eval)

Specified by:

finalScore in class BaseIEvaluationScoreCalculator<Model,RegressionEvaluation>