SHOGUN
v3.0.0
Main Page
Related Pages
Modules
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
src
shogun
evaluation
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
14
#include <
shogun/evaluation/BinaryClassEvaluation.h
>
15
16
namespace
shogun
17
{
18
19
class
CLabels;
20
25
class
CPRCEvaluation
:
public
CBinaryClassEvaluation
26
{
27
public
:
29
CPRCEvaluation
() :
30
CBinaryClassEvaluation
(),
m_computed
(false)
31
{
32
m_PRC_graph
=
SGMatrix<float64_t>
();
33
m_thresholds
=
SGVector<float64_t>
();
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
50
inline
EEvaluationDirection
get_evaluation_direction
()
const
51
{
52
return
ED_MAXIMIZE
;
53
}
54
58
float64_t
get_auPRC
();
59
63
SGMatrix<float64_t>
get_PRC
();
64
68
SGVector<float64_t>
get_thresholds
();
69
70
protected
:
71
73
SGMatrix<float64_t>
m_PRC_graph
;
74
76
SGVector<float64_t>
m_thresholds
;
77
79
float64_t
m_auPRC
;
80
82
bool
m_computed
;
83
};
84
85
}
86
87
#endif
/* PRCEVALUATION_H_ */
SHOGUN
Machine Learning Toolbox - Documentation