SHOGUN
v1.1.0
Main Page
Related Pages
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
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_ROC_graph
(NULL),
36
m_auROC
(0.0),
m_ROC_length
(0),
m_computed
(false) {};
37
39
virtual
~CROCEvaluation
();
40
42
virtual
inline
const
char
*
get_name
()
const
{
return
"ROCEvaluation"
; };
43
49
virtual
float64_t
evaluate
(
CLabels
* predicted,
CLabels
* ground_truth);
50
51
inline
EEvaluationDirection
get_evaluation_direction
()
52
{
53
return
ED_MAXIMIZE
;
54
}
55
59
float64_t
get_auROC
();
60
64
SGMatrix<float64_t>
get_ROC
();
65
69
SGVector<float64_t>
get_thresholds
();
70
71
protected
:
72
74
float64_t
*
m_ROC_graph
;
75
77
float64_t
*
m_thresholds
;
78
80
float64_t
m_auROC
;
81
83
int32_t
m_ROC_length
;
84
86
bool
m_computed
;
87
};
88
89
}
90
91
#endif
/* ROCEVALUATION_H_ */
SHOGUN
Machine Learning Toolbox - Documentation