文章目录
- 【机器学习】决策树算法(ID3算法及C4.5算法)的理解和应用
- 1.决策树的介绍
- 1.1 ID3算法
- 1.1.1 算法核心
- 1.1.2 基本概念
- 1.1.3 算法过程
- 1.2 C4.5算法
- 1.2.1 算法核心
- 1.2.2 基础概念
- 1.2.3 算法过程
- 2.决策树分类实战
- 2.1 C++实现ID3算法和C4.5算法
- 2.1.1 安装Graphviz(可视化决策树)
- 2.2 Python下实现CART算法
【机器学习】决策树算法(ID3算法及C4.5算法)的理解和应用
1.决策树的介绍
概念: 决策树是机器学习中一种特别常见的算法。决策树是一个预测模型,它在已知各种情况发生的概率下,训练而成的一个学习器,代表着对象属性和对象值的一种映射关系。
决策树结构:
根结点: 代表想要预测的目标的属性
分支: 代表想要预测的目标的属性类别
叶节点: 代表预测结果
1.1 ID3算法
1.1.1 算法核心
在信息论中有一个理论,信息期望越小,信息增益越大。在ID3算法中我们的主要目的就是要找到信息增益最大的属性做为决策树的根结点来进行决策树的建立。
1.1.2 基本概念
这里我们拿一个例子(相亲满意程度预测)来解释:
身高 | 年龄 | 是否有房 | 工作单位 | 满意程度 |
175 | 偏大 | 有 | 企业 | 满意 |
178 | 偏小 | 无 | 政府 | 满意 |
180 | 偏大 | 无 | 企业 | 不满意 |
178 | 偏大 | 无 | 政府 | 不满意 |
180 | 偏小 | 有 | 企业 | 满意 |
这里满意程度为预测结果,身高,年龄,是否有房,工作单位为构建决策树的属性。
1.信息熵: 度量一个属性的信息量
2.条件熵: 表示在已知属性X的情况下它的属性类别Y的不确定性(需要求所有属性的条件熵)
- 求身高的条件熵
- 求年龄的条件熵
- 求是否有房的条件熵
- 求工作单位的条件熵
**3.信息增益:**信息熵与条件熵的差
1.1.3 算法过程
在求出各属性的信息增益后,将属性增益最大的属性做为决策树的根结点。然后至上向下递归建树。这里简单的距离介绍确定第一个根结点后的步骤。
比如信息增益最大的属性为年龄,它有两个属性类别(即决策树的两个分支)
属性类别为“偏大”时(即分支为“偏大”)
身高 | 年龄 | 是否有房 | 工作单位 | 满意程度 |
175 | 偏大 | 有 | 企业 | 满意 |
178 | 偏大 | 无 | 企业 | 不满意 |
180 | 偏大 | 无 | 政府 | 不满意 |
属性类别为“偏小”时(即分支为“偏小”)
身高 | 年龄 | 是否有房 | 工作单位 | 满意程度 |
178 | 偏小 | 无 | 政府 | 满意 |
180 | 偏小 | 有 | 企业 | 满意 |
在此基础上在通过1.1.2中的基本概念求出各属性的信息增益,求出最大的信息增益在确定为根结点。
这里假设偏大分支下信息增益最大的属性是身高,偏小分支下信息增益最大的属性是工作单位。(如图)
1.2 C4.5算法
1.2.1 算法核心
在信息增益的基础上引入分裂率的概念,从而决定决策树根结点的因素变成了信息增益率。
1.2.2 基础概念
分裂率:
信息增益率:
其余概念与ID3算法相同。
1.2.3 算法过程
除了判断根结点属性的依据是信息增益率之外,其余过程与ID3算法相同。
2.决策树分类实战
2.1 C++实现ID3算法和C4.5算法
功能菜单:
数据集:
数据集可以自己制作,这里我是用网上常用的数据(偷个懒),但制作时注意进行预处理操作。
代码:
代码是在VS下编写的,我已经将代码的所有相关文件上传到Github上了,有兴趣的同学可以下载来试一下,除了功能6,需要安装配置Graphviz外(在下面有说明),其余功能可以直接实现。一共三个文件的decision_tree.cpp(主控源文件),pch.cpp(功能模块文件),pch.h(头文件)。
代码地址:https://github.com/huangzhaohong123/ID3-C4.5-decision_tree
2.1.1 安装Graphviz(可视化决策树)
DOT语言是用来描述图形的一种语言,而Graphviz则是用来处理这种语言的工具,适用于树结构的可视化。
该地址下下载graphviz-2.38.msi
https://graphviz.gitlab.io/_pages/Download/Download_windows.html 下载后会出现
然后一直点next,直到出现安装路径时,它会给你默认安装路径(这里个人建议放在你自己熟悉的路径下,方便后面配置环境变量)
安装后
环境变量配置:
搜索—>编辑系统环境变量—>高级—>环境变量—>系统变量中找到path—>添加安装路径加上bin
有的电脑需要重启一下,使得环境变量生效。
bin文件下标注的gvedit可以直接调用该工具
注意: 该工具在c++环境下不够友好,基本每次运行只能调用一次。
2.2 Python下实现CART算法
- 1.安装Graphviz(如上)
- 2.pip install graphviz
- 3.pip install pydotplus
数据集(配置适合你的隐形眼镜)
# -*- coding: UTF-8 -*-
from sklearn.preprocessing import LabelEncoder, OneHotEncoder
from sklearn.externals.six import StringIO
from sklearn import tree
import pandas as pd
import numpy as np
import pydotplus
import os
if __name__ == '__main__':
os.environ["PATH"] += os.pathsep + 'E:/Graphviz-2.38/bin' #若环境配置不成功,则直接代码调用环境,E:/Graphviz-2.38/bin为我的安装路径
with open('D:/desktop/experience/lenses1.txt', 'r') as fr: #加载文件
lenses = [inst.strip().split('\t') for inst in fr.readlines()] #处理文件
#print(lenses)
lenses_target = [] #提取每组数据的类别,保存在列表里
for each in lenses:
#print(each[-1])
lenses_target.append(each[-1])
#print(lenses_target)
lensesLabels = ['age', 'prescript', 'astigmatic', 'tearRate'] #特征标签
lenses_list = [] #保存lenses数据的临时列表
lenses_dict = {} #保存lenses数据的字典,用于生成pandas
for each_label in lensesLabels: #提取信息,生成字典
for each in lenses:
lenses_list.append(each[lensesLabels.index(each_label)])
lenses_dict[each_label] = lenses_list
lenses_list = []
#print(lenses_dict) #打印字典信息
lenses_pd = pd.DataFrame(lenses_dict) #生成pandas.DataFrame
print(lenses_pd) #打印pandas.DataFrame
le = LabelEncoder() #创建LabelEncoder()对象,用于序列化
for col in lenses_pd.columns: #序列化
lenses_pd[col] = le.fit_transform(lenses_pd[col])
print(lenses_pd) #打印编码信息
clf = tree.DecisionTreeClassifier(max_depth = 4) #创建DecisionTreeClassifier()类
clf = clf.fit(lenses_pd.values.tolist(), lenses_target) #使用数据,构建决策树
dot_data = StringIO()
tree.export_graphviz(clf, out_file = dot_data, #绘制决策树
feature_names = lenses_pd.keys(),
class_names = clf.classes_,
filled=True, rounded=True,
special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("tree1.pdf") #保存绘制好的决策树,以PDF的形式存储。
决策树可视化: