The class PRCEvalution used to evaluate PRC (Precision Recall Curve) graph of binary classifier. This class also has an capability of calculating auPRC (area under PRC).
Definition at line 27 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 () |
Protected Attributes | |
float64_t * | m_PRC_graph |
float64_t | m_auPRC |
int32_t | m_PRC_length |
bool | m_computed |
CPRCEvaluation | ( | ) |
constructor
Definition at line 31 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 48 of file PRCEvaluation.h.
virtual const char* get_name | ( | void | ) | const [virtual] |
area under PRC graph
Definition at line 69 of file PRCEvaluation.h.
bool m_computed [protected] |
indicator of PRC and auPRC being computed already
Definition at line 75 of file PRCEvaluation.h.
float64_t* m_PRC_graph [protected] |
2-d array used to store PRC graph
Definition at line 66 of file PRCEvaluation.h.
int32_t m_PRC_length [protected] |
number of points in PRC graph
Definition at line 72 of file PRCEvaluation.h.