SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
multiclass
tree
ID3ClassifierTree.h
浏览该文件的文档.
1
/*
2
* Copyright (c) The Shogun Machine Learning Toolbox
3
* Written (w) 2014 Parijat Mazumdar
4
* All rights reserved.
5
*
6
* Redistribution and use in source and binary forms, with or without
7
* modification, are permitted provided that the following conditions are met:
8
*
9
* 1. Redistributions of source code must retain the above copyright notice, this
10
* list of conditions and the following disclaimer.
11
* 2. Redistributions in binary form must reproduce the above copyright notice,
12
* this list of conditions and the following disclaimer in the documentation
13
* and/or other materials provided with the distribution.
14
*
15
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
16
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
19
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25
*
26
* The views and conclusions contained in the software and documentation are those
27
* of the authors and should not be interpreted as representing official policies,
28
* either expressed or implied, of the Shogun Development Team.
29
*/
30
31
32
#ifndef _ID3CLASSIFIERTREE_H__
33
#define _ID3CLASSIFIERTREE_H__
34
35
#include <
shogun/multiclass/tree/TreeMachine.h
>
36
#include <
shogun/multiclass/tree/ID3TreeNodeData.h
>
37
#include <
shogun/features/DenseFeatures.h
>
38
#include <
shogun/labels/MulticlassLabels.h
>
39
40
namespace
shogun
41
{
42
69
class
CID3ClassifierTree
:
public
CTreeMachine
<id3TreeNodeData>
70
{
71
public
:
73
CID3ClassifierTree
();
74
76
virtual
~CID3ClassifierTree
();
77
81
virtual
const
char
*
get_name
()
const
{
return
"ID3ClassifierTree"
; }
82
86
virtual
CMulticlassLabels
*
apply_multiclass
(
CFeatures
* data=NULL);
87
88
protected
:
89
93
virtual
bool
train_machine
(
CFeatures
* data=NULL);
94
95
private
:
96
102
node_t
* id3train(
CFeatures
* data,
CMulticlassLabels
* class_labels,
SGVector<int32_t>
values,
int
level = 0);
103
111
float64_t
informational_gain_attribute(int32_t attr_no,
CFeatures
* data,
CMulticlassLabels
* class_labels);
112
118
float64_t
entropy(
CMulticlassLabels
* labels);
119
120
};
121
}
/* namespace shogun */
122
123
#endif
/* _ID3CLASSIFIERTREE_H__ */
SHOGUN
机器学习工具包 - 项目文档