Class PRCEvaluation used to evaluate PRC (Precision Recall Curve) and an area under PRC curve (auPRC).
Definition at line 25 of file PRCEvaluation.h.
Public Member Functions | |
CPRCEvaluation () | |
virtual | ~CPRCEvaluation () |
virtual const char * | get_name () const |
virtual float64_t | evaluate (CLabels *predicted, CLabels *ground_truth) |
EEvaluationDirection | get_evaluation_direction () |
float64_t | get_auPRC () |
SGMatrix< float64_t > | get_PRC () |
SGVector< float64_t > | get_thresholds () |
Protected Attributes | |
float64_t * | m_PRC_graph |
float64_t * | m_thresholds |
float64_t | m_auPRC |
int32_t | m_PRC_length |
bool | m_computed |
CPRCEvaluation | ( | ) |
constructor
Definition at line 29 of file PRCEvaluation.h.
~CPRCEvaluation | ( | ) | [virtual] |
destructor
Definition at line 16 of file PRCEvaluation.cpp.
evaluate PRC and auPRC
predicted | labels | |
ground_truth | labels assumed to be correct |
Implements CBinaryClassEvaluation.
Definition at line 21 of file PRCEvaluation.cpp.
float64_t get_auPRC | ( | ) |
EEvaluationDirection get_evaluation_direction | ( | ) | [virtual] |
Implements CEvaluation.
Definition at line 46 of file PRCEvaluation.h.
virtual const char* get_name | ( | ) | const [virtual] |
get thresholds corresponding to points on the PRC graph
Definition at line 100 of file PRCEvaluation.cpp.
area under PRC graph
Definition at line 74 of file PRCEvaluation.h.
bool m_computed [protected] |
indicator of PRC and auPRC being computed already
Definition at line 80 of file PRCEvaluation.h.
float64_t* m_PRC_graph [protected] |
2-d array used to store PRC graph
Definition at line 68 of file PRCEvaluation.h.
int32_t m_PRC_length [protected] |
number of points in PRC graph
Definition at line 77 of file PRCEvaluation.h.
float64_t* m_thresholds [protected] |
vector with thresholds corresponding to points on the PRC graph
Definition at line 71 of file PRCEvaluation.h.