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