SHOGUN  v3.0.1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PRCEvaluation.h
Go to the documentation of this file.
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License as published by
4  * the Free Software Foundation; either version 3 of the License, or
5  * (at your option) any later version.
6  *
7  * Written (W) 2011 Sergey Lisitsyn
8  * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
9  */
10 
11 #ifndef PRCEVALUATION_H_
12 #define PRCEVALUATION_H_
13 
15 
16 namespace shogun
17 {
18 
19 class CLabels;
20 
26 {
27 public:
31  {
34  m_auPRC = 0.0;
35  };
36 
38  virtual ~CPRCEvaluation();
39 
41  virtual const char* get_name() const { return "PRCEvaluation"; };
42 
48  virtual float64_t evaluate(CLabels* predicted, CLabels* ground_truth);
49 
51  {
52  return ED_MAXIMIZE;
53  }
54 
59 
64 
69 
70 protected:
71 
74 
77 
80 
82  bool m_computed;
83 };
84 
85 }
86 
87 #endif /* PRCEVALUATION_H_ */
SGMatrix< float64_t > get_PRC()
The class Labels models labels, i.e. class assignments of objects.
Definition: Labels.h:35
SGVector< float64_t > get_thresholds()
virtual float64_t evaluate(CLabels *predicted, CLabels *ground_truth)
virtual const char * get_name() const
Definition: PRCEvaluation.h:41
EEvaluationDirection get_evaluation_direction() const
Definition: PRCEvaluation.h:50
EEvaluationDirection
Definition: Evaluation.h:24
double float64_t
Definition: common.h:48
SGVector< float64_t > m_thresholds
Definition: PRCEvaluation.h:76
SGMatrix< float64_t > m_PRC_graph
Definition: PRCEvaluation.h:73
Class PRCEvaluation used to evaluate PRC (Precision Recall Curve) and an area under PRC curve (auPRC)...
Definition: PRCEvaluation.h:25
The class TwoClassEvaluation, a base class used to evaluate binary classification labels...

SHOGUN Machine Learning Toolbox - Documentation