SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
evaluation
ROCEvaluation.h
浏览该文件的文档.
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
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
()
const
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
机器学习工具包 - 项目文档