SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
multiclass
tree
TreeMachine.h
浏览该文件的文档.
1
/*
2
* Copyright (c) The Shogun Machine Learning Toolbox
3
* Written (w) 2012 Chiyuan Zhang
4
* Written (w) 2014 Parijat Mazumdar
5
* All rights reserved.
6
*
7
* Redistribution and use in source and binary forms, with or without
8
* modification, are permitted provided that the following conditions are met:
9
*
10
* 1. Redistributions of source code must retain the above copyright notice, this
11
* list of conditions and the following disclaimer.
12
* 2. Redistributions in binary form must reproduce the above copyright notice,
13
* this list of conditions and the following disclaimer in the documentation
14
* and/or other materials provided with the distribution.
15
*
16
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
20
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26
*
27
* The views and conclusions contained in the software and documentation are those
28
* of the authors and should not be interpreted as representing official policies,
29
* either expressed or implied, of the Shogun Development Team.
30
*/
31
32
#ifndef TREEMACHINE_H__
33
#define TREEMACHINE_H__
34
35
#include <
shogun/machine/BaseMulticlassMachine.h
>
36
#include <
shogun/multiclass/tree/TreeMachineNode.h
>
37
#include <
shogun/multiclass/tree/BinaryTreeMachineNode.h
>
38
39
namespace
shogun
40
{
41
46
template
<
class
T>
class
CTreeMachine
:
public
CBaseMulticlassMachine
47
{
48
public
:
50
typedef
CTreeMachineNode<T>
node_t
;
51
53
typedef
CBinaryTreeMachineNode<T>
bnode_t
;
54
56
CTreeMachine
() :
CBaseMulticlassMachine
()
57
{
58
m_root
=NULL;
59
SG_ADD
((
CSGObject
**)&
m_root
,
"m_root"
,
"tree structure"
,
MS_NOT_AVAILABLE
);
60
}
61
63
virtual
~CTreeMachine
()
64
{
65
SG_UNREF
(
m_root
);
66
}
67
71
virtual
const
char
*
get_name
()
const
{
return
"TreeMachine"
; }
72
76
void
set_root
(
CTreeMachineNode<T>
* root)
77
{
78
SG_UNREF
(
m_root
);
79
SG_REF
(root);
80
m_root
=root;
81
}
82
86
CTreeMachineNode<T>
*
get_root
()
87
{
88
SG_REF
(
m_root
);
89
return
m_root
;
90
}
91
92
protected
:
94
CTreeMachineNode<T>
*
m_root
;
95
};
96
97
}
/* namespace shogun */
98
99
#endif
/* end of include guard: TREEMACHINE_H__ */
100
SHOGUN
机器学习工具包 - 项目文档