SHOGUN
v2.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
ROCEvaluation.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 ROCEVALUATION_H_
12
#define ROCEVALUATION_H_
13
14
#include <
shogun/evaluation/BinaryClassEvaluation.h
>
15
16
namespace
shogun
17
{
18
19
class
CLabels;
20
30
class
CROCEvaluation
:
public
CBinaryClassEvaluation
31
{
32
public
:
34
CROCEvaluation
() :
35
CBinaryClassEvaluation
(),
m_computed
(false)
36
{
37
m_ROC_graph
=
SGMatrix<float64_t>
();
38
m_thresholds
=
SGVector<float64_t>
();
39
};
40
42
virtual
~CROCEvaluation
();
43
45
virtual
inline
const
char
*
get_name
()
const
{
return
"ROCEvaluation"
; };
46
52
virtual
float64_t
evaluate
(
CLabels
* predicted,
CLabels
* ground_truth);
53
54
virtual
EEvaluationDirection
get_evaluation_direction
()
55
{
56
return
ED_MAXIMIZE
;
57
}
58
62
float64_t
get_auROC
();
63
67
SGMatrix<float64_t>
get_ROC
();
68
72
SGVector<float64_t>
get_thresholds
();
73
74
protected
:
75
81
float64_t
evaluate_roc
(
CLabels
* predicted,
CLabels
* ground_truth);
82
83
protected
:
84
86
SGMatrix<float64_t>
m_ROC_graph
;
87
89
SGVector<float64_t>
m_thresholds
;
90
92
float64_t
m_auROC
;
93
95
bool
m_computed
;
96
};
97
98
}
99
100
#endif
/* ROCEVALUATION_H_ */
SHOGUN
Machine Learning Toolbox - Documentation