Package org.deeplearning4j.earlystopping.scorecalc

Class ROCScoreCalculator

java.lang.Object

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

org.deeplearning4j.earlystopping.scorecalc.ROCScoreCalculator

All Implemented Interfaces:

Serializable, ScoreCalculator<Model>

public class ROCScoreCalculator
extends BaseIEvaluationScoreCalculator<Model,IEvaluation>

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ROCScoreCalculator.Metric
static class	ROCScoreCalculator.ROCType

Field Summary

Fields

Modifier and Type	Field	Description
protected ROCScoreCalculator.Metric	metric
protected ROCScoreCalculator.ROCType	type

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

iter, iterator

Constructor Summary

Constructors

Constructor	Description
ROCScoreCalculator(ROCScoreCalculator.ROCType type, ROCScoreCalculator.Metric metric, DataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, ROCScoreCalculator.Metric metric, MultiDataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, DataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, MultiDataSetIterator iterator)

Method Summary

Modifier and Type	Method	Description
protected double	finalScore(IEvaluation eval)
boolean	minimizeScore()
protected IEvaluation	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

type

protected final ROCScoreCalculator.ROCType type

metric

protected final ROCScoreCalculator.Metric metric

Constructor Detail

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          DataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          MultiDataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          ROCScoreCalculator.Metric metric,
                          DataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          ROCScoreCalculator.Metric metric,
                          MultiDataSetIterator iterator)

Method Detail

newEval

protected IEvaluation newEval()

Specified by:

newEval in class BaseIEvaluationScoreCalculator<Model,IEvaluation>

finalScore

protected double finalScore(IEvaluation eval)

Specified by:

finalScore in class BaseIEvaluationScoreCalculator<Model,IEvaluation>

minimizeScore

public boolean minimizeScore()

Returns:

If true: the score should be minimized. If false: the score should be maximized.