java id3_经典ID3算法

最新推荐文章于 2021-02-20 15:03:41 发布

群青色黑洞

最新推荐文章于 2021-02-20 15:03:41 发布

阅读量219

点赞数

文章标签： java id3

本文链接：https://blog.csdn.net/weixin_33646989/article/details/114422756

版权

1．背景知识：

决策树是对数据进行分类，以此达到预测的目的。该决策树方法先根据训练集数据形成决策树，如果该树不能对所有对象给出正确的分类，那么选择一些例外加入到

训练集数据中，重复该过程一直到形成正确的决策集。决策树代表着决策集的树形结构。

决策树由决策结点、分支和叶子组成。决策树中最上面的结点为根结点，每个分支是一个新的决策结点，或者是树的叶子。每个决策结点代表一个问题或决策，通常对应于待分类对象的属性。每一个叶子结点代表一种可能的分类结果。沿决策树从上到下遍历的过程中，在每个结点都会遇到一个测试，对每个结点上问题的不同的测试输出导致不同的分支，最后会到达一个叶子结点，这个过程就是利用决策树进行分类的过程，利用若干个变量来判断所属的类别。

2．ID3算法：

ID3算法是由Quinlan首先提出的。该算法是以信息论为基础，以信息熵和信息增益度为衡量标准，从而实现对数据的归纳分类。以下是一些信息论的基本概念：

定义1：若存在n个相同概率的消息，则每个消息的概率p是1/n，一个消息传递的信息量为Log2(n)

定义2：若有n个消息，其给定概率分布为P=(p1,p2…pn)，则由该分布传递的信息量称为P的熵，记为

I(p)=-(i=1 to n求和)piLog2(pi)。

定义3：若一个记录集合T根据类别属性的值被分成互相独立的类C1C2..Ck，则识别T的一个元素所属哪个类所需要的信息量为Info(T)=I(p)，其中P为C1C2…Ck的概率分布，即P=(|C1|/|T|,…..|Ck|/|T|)

定义4：若我们先根据非类别属性X的值将T分成集合T1,T2…Tn，则确定T中一个元素类的信息量可通过确定Ti的加权平均值来得到，即Info(Ti)的加权平均值为：

Info(X, T)=(i=1 to n 求和)((|Ti|/|T|)Info(Ti))

定义5：信息增益度是两个信息量之间的差值，其中一个信息量是需确定T的一个元素的信息量，另一个信息量是在已得到的属性X的值后需确定的T一个元素的信息量，信息增益度公式为：

Gain(X, T)=Info(T)-Info(X, T)

为方便理解，在此举例说明算法过程，具体算法其实很简单，呵呵：

某市高中一年级(共六个班)学生上学期期末考试成绩数据库。其中学生考试成绩属性有：学籍号、语文、数学、英语、物理、化学，如下表所示，本例子的目的是利用决策树技术研究学生物理成绩的及格与否可以由哪些属性决定

学号语文数学英语物理化学

1 76 71 68 71 81

2 71 65 63 72 74

3 60 81 67 87 80

4 67 84 71 61 78

5 64 76 72 64 72

6 73 80 66 58 67

7 62 81 78 52 79

8 74 78 47 60 56

9 62 68 63 74 67

10 72 73 73 49 60

…… …… …… …… …… ……

第一步：对数据进行规范化处理。

将上表中的数据规范化，用0表示成绩小于60分，1表示成绩大于或等于60分，得到下表：

学号语文数学英语物理化学

1 76 71 68 71 81

2 71 65 63 72 74

3 60 81 67 87 80

4 67 84 71 61 78

5 64 76 72 64 72

6 73 80 66 58 67

7 62 81 78 52 79

8 74 78 47 60 56

9 62 68 63 74 67

10 72 73 73 49 60

…… …… …… …… …… ……

第二步：选取训练实例集。

从所有学生中进行抽样，将抽样数据作为训练集，共计有161条记录。经统计，在这161条记录的训练集中单科成绩及格人数和不及格人数如下表所示：

语文数学英语物理化学

及格 82 57 34 32 39

不及格 79 104 127 129 122

第三步：利用信息增益度选取最能区别训练集中实例的属性。

首先计算课程物理所含有的信息量。由表4可知物理及格人数P=32，不及格人数N=129，则可得到：

Info(T)=I(32, 129)=-[(32/161)Log2(32/161)+(129/161)Log2(129/161)]=0.7195

然后计算当课程物理及格和不及格时，课程语文所包含的总信息量。经统计，语文和物理有如下表所示的统计数据：

成绩搭配人数

语文成绩=1且物理成绩=1 28

语文成绩=1且物理成绩=0 54

语文成绩=0且物理成绩=1 4

语文成绩=0且物理成绩=0 75

可得到：

Info(X,T) = )=(i=1 to n 求和)((|Ti|/|T|)Info(Ti))=(82/161)I(28,54)+(79/161)I(4,75)=0.6136

最后可得到语文的信息增益度为：

Gain(X,T)=Info(T)-Info(X,T)=0.7195-0.6136=0.1059

同理可得其他课程的信息增益度，结果如下表所示：

数学英语化学

Gain 0.2136 0.095 0.1701

由此可以看出所有课程当中数学是最能区别训练集中决定物理成绩与否的课程

第四步：创建一个树结点，并创建该结点的子链，每个子链代表所选属性的一个唯一值。使用子链的值进一步细化子类。当出现以下两种情形之一时可以停止分类：1.一个结点上的数据都是属于同一类别；2.没有属性可以再对属性进行分割。

根据各个课程的信息增益度，应该选择数学作为所建决策树的根结点。由于数学的属性值只有两个：1(及格)和0(不及格)，所以在数学下可以建立两个分支。经统计，数学不及格且物理不及格的人数为100，其准确率为100/104=96.2%。因此对数学不及格这个分之停止分割。又经统计，数学及格的57人中有26人物理及格，31人物理不及格，所以应对数学及格这个分支进行分割。从上表可知，应该选取化学作为分割结点进行细化。分割后经统计显示，数学和化学都及格的学生中，有26人物理及格，6人物理不及格，准确率为 26/32=81.3%；数学及格但化学不及格的学生中，有22人物理不及格，3人物理及格，准确率为 22/25=88%。由此可构建出数据的决策树，如下所示

数学

(及格) (不及格)

化学物理不及格(104/4)

(及格) (不及格)

物理及格(32/6) 物理不及格(25/3)

注：括号内为分支条件(不知道怎么上传图片，实际是一棵树，呵呵)

第五步：将其它成绩作为检验集。并用来检验所生成的决策树的准确度。

由该决策树可以得出下列规则：

(1)IF学生的数学成绩不及格

THEN其物理成绩通常也不及格。

准确度=(104-4)/104=96.2％

覆盖率=104/161=64.6％