SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
multiclass
tree
BalancedConditionalProbabilityTree.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) 2012 Chiyuan Zhang
8
* Copyright (C) 2012 Chiyuan Zhang
9
*/
10
11
#ifndef BALANCEDCONDITIONALPROBABILITYTREE_H__
12
#define BALANCEDCONDITIONALPROBABILITYTREE_H__
13
14
#include <
shogun/multiclass/tree/ConditionalProbabilityTree.h
>
15
16
namespace
shogun
17
{
18
35
class
CBalancedConditionalProbabilityTree
:
public
CConditionalProbabilityTree
36
{
37
public
:
39
CBalancedConditionalProbabilityTree
();
40
42
virtual
~CBalancedConditionalProbabilityTree
() {}
43
45
virtual
const
char
*
get_name
()
const
{
return
"BalancedConditionalProbabilityTree"
; }
46
48
void
set_alpha
(
float64_t
alpha);
49
51
float64_t
get_alpha
()
const
{
return
m_alpha; }
52
53
protected
:
59
virtual
bool
which_subtree
(
bnode_t
*
node
,
SGVector<float32_t>
ex);
60
61
private
:
66
int32_t tree_depth(
bnode_t
*
node
);
67
69
float64_t
m_alpha;
70
};
71
72
}
/* shogun */
73
74
#endif
/* end of include guard: BALANCEDCONDITIONALPROBABILITYTREE_H__ */
75
SHOGUN
机器学习工具包 - 项目文档