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>

Calculate ROC AUC (area under ROC curve) or AUCPR (area under precision recall curve) for a MultiLayerNetwork or ComputationGraph

Author:

Alex Black

See Also:

Serialized Form

Nested Class Summary

Nested Classes

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

Field Summary

Fields

Modifier and Type	Field and 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 and Description
ROCScoreCalculator(ROCScoreCalculator.ROCType type, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, ROCScoreCalculator.Metric metric, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
ROCScoreCalculator(ROCScoreCalculator.ROCType type, ROCScoreCalculator.Metric metric, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

Method Summary

Modifier and Type	Method and Description
protected double	finalScore(IEvaluation eval)
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,
                          org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          ROCScoreCalculator.Metric metric,
                          org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

ROCScoreCalculator

public ROCScoreCalculator(ROCScoreCalculator.ROCType type,
                          ROCScoreCalculator.Metric metric,
                          org.nd4j.linalg.dataset.api.iterator.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>