Package org.nd4j.evaluation.curves

Class RocCurve

java.lang.Object

org.nd4j.evaluation.curves.BaseCurve

org.nd4j.evaluation.curves.RocCurve

public class RocCurve
extends BaseCurve

Field Summary

Fields inherited from class org.nd4j.evaluation.curves.BaseCurve

DEFAULT_FORMAT_PREC

Constructor Summary

Constructors

Constructor	Description
RocCurve(double[] threshold, double[] fpr, double[] tpr)

Method Summary

Modifier and Type	Method	Description
double	calculateAUC()	Calculate and return the area under ROC curve
static RocCurve	fromJson(String json)
static RocCurve	fromYaml(String yaml)
double	getFalsePositiveRate(int i)
double	getThreshold(int i)
String	getTitle()
double	getTruePositiveRate(int i)
double[]	getX()
double[]	getY()
int	numPoints()

Methods inherited from class org.nd4j.evaluation.curves.BaseCurve

calculateArea, calculateArea, format, fromJson, fromYaml, toJson, toYaml

Methods inherited from class java.lang.Object

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

Constructor Detail

RocCurve

public RocCurve(double[] threshold,
                double[] fpr,
                double[] tpr)

Method Detail

numPoints

public int numPoints()

Specified by:

numPoints in class BaseCurve

Returns:

The number of points in the curve

getX

public double[] getX()

Specified by:

getX in class BaseCurve

Returns:

X axis values

getY

public double[] getY()

Specified by:

getY in class BaseCurve

Returns:

Y-axis values

getTitle

public String getTitle()

Specified by:

getTitle in class BaseCurve

Returns:

Title for the curve

getThreshold

public double getThreshold(int i)

Parameters:

i - Point number, 0 to numPoints()-1 inclusive

Returns:

Threshold of a given point

getTruePositiveRate

public double getTruePositiveRate(int i)

Parameters:

i - Point number, 0 to numPoints()-1 inclusive

Returns:

True positive rate of a given point

getFalsePositiveRate

public double getFalsePositiveRate(int i)

Parameters:

i - Point number, 0 to numPoints()-1 inclusive

Returns:

False positive rate of a given point

calculateAUC

public double calculateAUC()

Calculate and return the area under ROC curve

fromJson

public static RocCurve fromJson(String json)

fromYaml

public static RocCurve fromYaml(String yaml)