1. 决策树原理

1.1. 特性

决策树属于非参数监督学习方法，在学习数据特征推断出简单的决策规划创建一个预测目标变量值的模型。

优点

1. 可解释型强，而且决策树很容易可视化。此外，决策树使用“白箱模型”，比起黑箱模型，解释性也更强。
2. 能处理多输出问题。
3. 即使其假设与生成数据的真实模型有一定程度的违背，也能很好地执行。
4. 使用树（即预测数据）的成本在用于训练树的数据点的数量上是对数的。

缺点

1. 很容易就过拟合了。所以才需要剪枝等技巧或者设置叶节点所需最小样本树或者设置树最大深度。
2. 不稳定，因为时递归出来的，很微小的变化可能导致整个树结构大变。
3. 不擅长泛化和外推。因为决策树的预测不是平滑的也不是连续的，所以中间的空挡很难处理，所以在外推方面表现差。
4. 决策树基于的启发式算法可能会导致产生的决策树并非最优决策树，比如在使用贪心算法的时候。因此才会有人提出特征和样本需要被随机抽样替换。
5. 在一些问题上先天不足。如：XOR、奇偶校验或多路复用器问题。
6. 不支持缺失值，这点在随机森林中也有所提现，所以采取本方法时注意缺失值处理或填补。

1.2. 思路

在计算机中，决策树更多的时用于分类。有点类似于用 if else 语句去穷举每一种可能，但是面对多个判断的情况时，每个判断出现的位置还不确定，我们需要通过某种方法确定各个条件出现顺序（谁在树根谁在下面），这个方法是决策树的一个难点。此外，面对大量数据时，建立的决策树模型可能会特别特别大，深度深，叶子节点多且节点中的样例少等可能性，优化这个问题需要的方法也是一个难题。

所有的决策树的步骤的最终目的都是希望通过学习产生一个泛化能力强的决策树，即处理未见示例能力强的决策树。

1.3. 概念

决策树概念

根据西瓜书的定义：⼀棵决策树包含⼀个根节点，若⼲个内部节点和若⼲个叶⼦节点。叶结点对应决策树决策结果，其他每个结点则对应于⼀个属性测试。

信息论

决策树这里用到了信息论中间很基础的知识，我们希望决策树的分支节点所包含的样本尽可能都属于一个类别，那么这么衡量这个？这需要用到纯度。直观上讲就是分类纯度越低，里面的标签就越杂 。

那么纯度怎么表示？

可以用信息熵表示。假定当前样本集合 D 中第 k 类样本所占的比例为 $P_k$ (k =1, 2,. . . ,$|\gamma |$) ，则D的信息熵定义为:
$$Ent(D)=-\sum _{k=1}^{|\gamma |}{p}_{k}lo{g}_2{p}_k$$

Ent(D)的值越小，则D的纯度越⾼。
Ent(D)的值越大，则D的不纯度越大，则D的不确定性越大。

2. 公式推导

决策树模型构建一般分为三步，

1. 特征选择
2. 决策树生成
3. 决策树修剪

特征选择更多的是取决于数据集，在生活中的问题里可能更加重要，一般来说没有什么特定的标准，需要注意的点比较琐碎，这边不细讲。

决策树生成和修剪的难度更大，这里做出总结。

2.1. 构造决策树

如何构造决策树其实就是如何选取每层节点的问题，西瓜书上介绍了三种方法，这边也按照这三种方法总结，其他的方法我将在面试题中提到。

2.1.1. ID3

理论

靠信息增益区分。

我们知道信息熵可以确定经过一个节点划分后每个分支节点的纯度，但是我们这里需要确定节点的顺序，所以单单知道划分后每个分支节点的纯度还不够，应该通过某种方法利用分支节点的纯度反映到划分节点的优劣。

ID3就是使用信息增益来判断每个节点的划分能力的。信息增益公式如下：

$$Gain(D,\alpha )=Ent(D)-\sum _{v=1}^V\frac{|D^v|}{|D|}Ent(D^v)$$

$\frac{|D^v|}{|D|}$就是每个分支节点的划分权重，所以样本数量越多划分权重越大。$Gain(D,\alpha )$就是使用$\alpha$属性划分对整体纯度提升的能力。因此我们有 ID3 的逻辑顺序：

• 从根节点开始，会节点计算所有可能的特征的信息增益，选择信息增益最大的特征作为节点的特征。
• 由该特征的不同取值建立子节点，在对子节点递归地调用上面方法，构建决策树，知道所有的特征信息增益都很小或者没有特征可以选为止。
• 得到决策树模型。

示例

西瓜数据集

编号	色泽	根蒂	敲声	纹理	脐部	触感	密度	含糖率	好瓜
1	0	0	0	0	0	0	0.697	0.46	1
2	1	0	1	0	0	0	0.774	0.376	1
3	1	0	0	0	0	0	0.634	0.264	1
4	0	0	1	0	0	0	0.608	0.318	1
5	2	0	0	0	0	0	0.556	0.215	1
6	0	1	0	0	1	1	0.403	0.237	1
7	1	1	0	1	1	1	0.481	0.149	1
8	1	1	0	0	1	0	0.437	0.211	1
9	1	1	1	1	1	0	0.666	0.091	0
10	0	2	2	0	2	1	0.243	0.267	0
11	2	2	2	2	2	0	0.245	0.057	0
12	2	0	0	2	2	1	0.343	0.099	0
13	0	1	0	1	0	0	0.639	0.161	0
14	2	1	1	1	0	0	0.657	0.198	0
15	1	1	0	0	1	1	0.36	0.37	0
16	2	0	0	2	2	0	0.593	0.042	0
17	0	0	1	1	1	0	0.719	0.103	0

计算根蒂的信息增益：

缺点

对种类多的属性有偏好，比如编号属性。这个属性的信息增益会特别高，但是实际上它所包含的信息没多少。

此时就需要另一个算法 C4.5

2.1.2. C4.5

理论

靠增益率区分。

既然考虑到信息增益对种类多的属性的偏好，增益率应该削弱这类属性的权重。于是提出增益的权重，也称为属性的固有值。

对信息增益提出了惩罚，特征可取值数量较多时，惩罚参数大；特征可取值数量较少时，惩罚参数小。

$$IV(\alpha )=-\sum _{v=1}^V\frac{|D^v|}{|D|}lo{g}_2\frac{|D^v|}{|D|}$$
$$Gain\_ratio=\frac{Gain(D,\alpha )}{IV(\alpha )}$$

底层思路是讲连续特征离散化，所有也可以用来处理连续型数据。

示例

西瓜数据集

编号	色泽	根蒂	敲声	纹理	脐部	触感	密度	含糖率	好瓜
1	0	0	0	0	0	0	0.697	0.46	1
2	1	0	1	0	0	0	0.774	0.376	1
3	1	0	0	0	0	0	0.634	0.264	1
4	0	0	1	0	0	0	0.608	0.318	1
5	2	0	0	0	0	0	0.556	0.215	1
6	0	1	0	0	1	1	0.403	0.237	1
7	1	1	0	1	1	1	0.481	0.149	1
8	1	1	0	0	1	0	0.437	0.211	1
9	1	1	1	1	1	0	0.666	0.091	0
10	0	2	2	0	2	1	0.243	0.267	0
11	2	2	2	2	2	0	0.245	0.057	0
12	2	0	0	2	2	1	0.343	0.099	0
13	0	1	0	1	0	0	0.639	0.161	0
14	2	1	1	1	0	0	0.657	0.198	0
15	1	1	0	0	1	1	0.36	0.37	0
16	2	0	0	2	2	0	0.593	0.042	0
17	0	0	1	1	1	0	0.719	0.103	0

计算根蒂的信息增益率：

缺点

缺点也很明显，就是矫枉过正。面对可取数值比较少的属性时，增益率存在偏好。因此有了基指系数。

2.1.3. CART

靠基尼指数划分。

我们可以理解为随机抽取两个样本，其类别不一样的概率的加权和。这个权重跟 C4.5 的权重划分方法相同。

这个本质上是一个回归树，对于计算机来说虽然很多时候结果跟信息熵算出来的差不多，但是计算会快很多，因为没有对数。

数据集的基尼指数：

$$Gini(D)=\sum _{k=1}^{|\gamma |}\sum _{k^{{}^{\prime }}\ne k}{p}_k{p}_{k^{{}^{\prime }}}=1-\sum _{k=1}^{|\gamma |}{p}_k^2=1-\sum _{k=1}^{|\gamma |}(\frac{|D_i|}{|D|}{)}^2$$

数据集纯度越高，${\sum }_{k=1}^{|\gamma |}{p}_k^2$ 越大，Gini 越小。

属性的基尼指数：

$$Gini\_index(D,\alpha )=\sum _{v=1}^V\frac{|D^v|}{|D|}Gini(D^v)$$

最优划分属性就是 $Gini\_index(D,\alpha )$ 最小的属性。

示例

西瓜数据集

编号	色泽	根蒂	敲声	纹理	脐部	触感	密度	含糖率	好瓜
1	0	0	0	0	0	0	0.697	0.46	1
2	1	0	1	0	0	0	0.774	0.376	1
3	1	0	0	0	0	0	0.634	0.264	1
4	0	0	1	0	0	0	0.608	0.318	1
5	2	0	0	0	0	0	0.556	0.215	1
6	0	1	0	0	1	1	0.403	0.237	1
7	1	1	0	1	1	1	0.481	0.149	1
8	1	1	0	0	1	0	0.437	0.211	1
9	1	1	1	1	1	0	0.666	0.091	0
10	0	2	2	0	2	1	0.243	0.267	0
11	2	2	2	2	2	0	0.245	0.057	0
12	2	0	0	2	2	1	0.343	0.099	0
13	0	1	0	1	0	0	0.639	0.161	0
14	2	1	1	1	0	0	0.657	0.198	0
15	1	1	0	0	1	1	0.36	0.37	0
16	2	0	0	2	2	0	0.593	0.042	0
17	0	0	1	1	1	0	0.719	0.103	0

计算根蒂的基尼指数：

对比分析

2.2. 剪枝

剪枝 (pruning)是决策树学习算法对付"过拟合"的主要⼿段。在决策树学习中，为了尽可能正确分类训练样本，结点划分过程将不断重复，有时会造成决策树分支过多，这时就可能因训练样本学得"太好"了，以致于把训练集自身的⼀些特点当作所有数据都具有的⼀般性质⽽导致过拟合。

决策树剪枝的基本策略有“预剪枝” (prepruning)和“后剪枝 ”(post pruning) [Quinlan, 1993]. 预剪枝是指在决策树⽣成过程中，对每个结点在划分前先进⾏估计，若当前结点的划分不能带来决策树泛化性能提升，则停⽌划分并将当前结点标记为叶结点；后剪枝则是先从训练集⽣成⼀棵完整的决策树， 然后自底向上地对非叶结点进⾏考察，若将该结点对应的⼦树替换为叶结点能带来决策树泛化性能提升，则将该⼦树替换为叶结点。

3. 实例

3.1. 数据集

watermelon.csv

x1,x2,x3,x4,x5,x6,label
0,0,0,0,0,0,1
1,0,1,0,0,0,1
1,0,0,0,0,0,1
0,0,1,0,0,0,1
2,0,0,0,0,0,1
0,1,0,0,1,1,1
1,1,0,1,1,1,1
1,1,0,0,1,0,1
1,1,1,1,1,0,0
0,2,2,0,2,1,0
2,2,2,2,2,0,0
2,0,0,2,2,1,0
0,1,0,1,0,0,0
2,1,1,1,0,0,0
1,1,0,0,1,1,0
2,0,0,2,2,0,0
0,0,1,1,1,0,0

3.2. ID3

# 计算信息熵
def c_entropy(label):# 每类有几个count_class = Counter(label)ret = 0for classes in count_class.keys():ret += count_class[classes]/len(label)*np.log2(count_class[classes]/len(label))return -ret# 计算单个属性信息增益
def c_entropy_gain(data, label):ent = c_entropy(label)# 几个值循环几次count_value = Counter(data)sum_entropy = 0for value_a in count_value.keys():sum_entropy += count_value[value_a] / len(label) * c_entropy(label[data == value_a])return ent - sum_entropy# 计算计算每个属性的信息增益并求最大的
def find_entropy_index(data, label):label = label.flatten()feature_count = data.shape[1]max_list = []for i in range(feature_count):max_list.append(c_entropy_gain(data[:, i], label))max_value = max(max_list)  # 求列表最大值max_idx = max_list.index(max_value)  # 求最大值对应索引return max_idx

3.3. C4.5

# 计算信息熵
def c_entropy(label):# 每类有几个count_class = Counter(label)ret = 0for classes in count_class.keys():ret += count_class[classes]/len(label)*np.log2(count_class[classes]/len(label))return -ret# 计算单个属性信息增益率
def c_entropy_gain_rate(data, label):ent = c_entropy(label)# 几个值循环几次count_value = Counter(data)sum_entropy = 0sum_iv = 0for value_a in count_value.keys():d = count_value[value_a] / len(label)sum_entropy += d * c_entropy(label[data == value_a])sum_iv -= d * np.log2(d)return (ent - sum_entropy) / sum_iv# 计算计算每个属性的信息增益率并求最大的
def find_entropy_ratio(data, label):# 返回大于中位数的下标def get_mid_index(list):mid = int(len(list)/2)b = sorted(enumerate(list), key=lambda x: x[1])  # x[1]是因为在enumerate(a)中，a数值在第1位c = [x[0] for x in b]  # 获取排序好后b坐标,下标在第0位return c[mid:]label = label.flatten()feature_count = data.shape[1]max_list_e = []max_list = []for i in range(feature_count):max_list_e.append(c_entropy_gain(data[:, i], label))max_list.append(c_entropy_gain_rate(data[:, i], label))# 只保留信息增益大于平均水平的max_list_mid = [max_list[i] for i in get_mid_index(max_list_e)]max_value = max(max_list_mid)  # 求列表最大值max_idx = max_list_mid.index(max_value)  # 求最大值对应索引return max_idx

3.4. CART

# 计算单个属性基尼指数
def c_gini(data, label):# 几个值循环几次count_value = Counter(data)gini = 0for value_a in count_value.keys():d = count_value[value_a] / len(label)class_gini = 1# 取出值一样的data的valuesame_value_class = label[data == value_a]count_labels = Counter(same_value_class)for i in count_labels.values():class_gini -= (i / count_value[value_a])**2gini += d * class_ginireturn gini# 计算计算每个属性的基尼指数并求最小的
def find_gini(data, label):label = label.flatten()feature_count = data.shape[1]min_list = []for i in range(feature_count):min_list.append(c_gini(data[:, i], label))min_value = min(min_list)  # 求列表最小值min_idx = min_list.index(min_value)  # 求最小值对应索引return min_idx

3.5. sklearn实现

def sk(data_train, label_train, data_test, label_test):clf = tree.DecisionTreeClassifier(criterion="entropy")clf = clf.fit(data_train, label_train)score = clf.score(data_test, label_test)  # 返回精确度print("准确率为：", score * 100, "%")# 画决策树feature_name = ['色泽', '根蒂', '敲声', '纹理', '脐部', '触感']import matplotlib as mplmpl.rcParams['font.sans-serif'] = ['FangSong']  # 指定中文字体mpl.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题plt.rcParams['font.sans-serif'] = ['SimHei']plt.rcParams['axes.unicode_minus'] = False  # 正常显示负号fig, axes = plt.subplots(nrows=1, ncols=1, figsize=(4, 4), dpi=300)tree.plot_tree(clf,feature_names=feature_name,class_names=["好瓜", "坏瓜"],filled=True,  # 填充颜色，颜色越深，不纯度越低rounded=True  # 框的形状)plt.show()

结果如下

可视化一下：

4. 几个注意点(面试问题)

  决策树有几个注意点，可能会在面试中被提到，还是比较能体现被面试者对这个算法的理解的。当然我们也不一定就是为了面试，搞清楚这些问题对帮助我们理解这个算法还是很有好处的。其中部分答案是博主自己的理解，如果有问题麻烦路过的大佬评论区指正。

1. 决策树常使用那些常用的启发函数？

答：

• ID3 => 最大信息增益
• C4.5 => 最大信息增益比
• CART => 最大基尼指数

这仨还有一些特点：

• ID3 泛化能力弱，C4.5一定程度上惩罚了取值较多的特征，避免ID3的过拟合，提升泛化能力。

• ID3 只能处理离散数据，其他俩都可以处理连续型变量，C4.5靠排序后找分割线把连续型变量转化为布尔型，CART靠对特征二值划分。

• ID3 C4.5 只能做分类任务，CART可以分类也可以回归。

• ID3 对缺失值特别敏感。

• ID3 C4.5 每个节点可以产生多分支，CART只会产生二分支。

2. 决策树是如何进行剪枝的？

答：

老生常谈的问题了，前面也有讲。这里就重复讲下。

预剪枝

在树长到一定深度后停止生长，当树的节点样本数小于某个阈值时停止生长，每次分裂树带来的提升小于某个阈值时停止生长。

可能导致欠拟合。不过比较适合大规模的问题，在分类后期准确率会有显著增长。需要有经验的人确定相关参数（节点样本数、最大深度等）

后剪枝

后剪枝是在树建立后，从下往上计算是否需要剪掉一部分子树并用叶子代替。比较难的地方是几个后剪枝的方法（错误率降低后剪枝、悲观剪枝、代价复杂度剪枝、最小误差剪枝、CVP、OPP 等）。这个比较复杂，不详细介绍。

3. 决策树递归时导致递归返回的三种条件？

答：

• 当前节点包含的样本全部属于统同一类别，无需划分。
• 当前属性集为空，或是所有样本在所有属性上取值相同，无法划分。
• 当前节点包含的样本集合为空，不能划分。

4. 计算恋爱数据集。数据集如下：

样本	颜值（A）	身材（B）	性格（C）	收入（D）	学历（E）	交往（R）
1	1	0	1	1	1	1
2	1	1	0	0	0	0
3	1	1	1	1	1	1
4	1	0	1	0	1	0
5	1	1	0	1	0	0
6	0	1	0	1	1	0
7	1	1	0	0	0	0
8	1	0	1	1	1	1
9	0	0	1	1	1	0
10	1	1	1	1	1	1
11	1	0	1	0	0	0
12	0	0	1	1	1	0
13	1	1	1	0	1	1
14	1	0	1	1	1	1
15	1	1	1	1	1	1

答：
暂时就算一个颜值和身材哈~~

5. 运行（可直接食用）

import random
from collections import Counter
import numpy as np
import pandas as pd
import warnings
from matplotlib import pyplot as plt
warnings.filterwarnings("ignore")
from sklearn import treeclass Node(object):def __init__(self, son=None, data=None, label=None, node_label=None):self.son = son      # 节点的子节点self.data = data    # 节点包含的数据self.label = label  # 节点包含的标签self.node_label = node_label    # 按照哪个属性分类self.if_leaf = False        # 默认不是子节点def build_tree(data, label, choice):mytree = Node()# 全部属于一个类别if np.unique(label.flatten()).shape[0] == 1:mytree.if_leaf = Truemytree.label = label[0]return mytree# 当前属性都属于一个类别if np.unique(data, axis=0).shape[0] == 1:mytree.if_leaf = True# 听天由命，直接选第一个mytree.label = label[0]return mytree# 样本或属性集合是空的if label.shape[0] == 0:mytree.if_leaf = True# 从来没碰到这种情况，默认0mytree.label = 0return mytree# 判断if choice == 1:mytree = Node(data=data, label=label, node_label=find_entropy_index(data, label))if choice == 2:mytree = Node(data=data, label=label, node_label=find_entropy_ratio(data, label))if choice == 3:mytree = Node(data=data, label=label, node_label=find_gini(data, label))mytree.son_label = Counter(data[:, mytree.node_label]).keys()# 递归# for feature_class in Counter(data[:, mytree.node_label]).keys():#     print(data[data[:, mytree.node_label] == feature_class], "\n", label[data[:, mytree.node_label] == feature_class],"\n\n")mytree.son = {feature_class: build_tree(data[data[:, mytree.node_label] == feature_class], label[data[:, mytree.node_label] == feature_class], choice) for feature_class in Counter(data[:, mytree.node_label]).keys()}return mytree# 计算信息熵
def c_entropy(label):# 每类有几个count_class = Counter(label)ret = 0for classes in count_class.keys():ret += count_class[classes]/len(label)*np.log2(count_class[classes]/len(label))return -ret# 计算单个属性信息增益
def c_entropy_gain(data, label):ent = c_entropy(label)# 几个值循环几次count_value = Counter(data)sum_entropy = 0for value_a in count_value.keys():sum_entropy += count_value[value_a] / len(label) * c_entropy(label[data == value_a])return ent - sum_entropy# 计算计算每个属性的信息增益并求最大的
def find_entropy_index(data, label):label = label.flatten()feature_count = data.shape[1]max_list = []for i in range(feature_count):max_list.append(c_entropy_gain(data[:, i], label))max_value = max(max_list)  # 求列表最大值max_idx = max_list.index(max_value)  # 求最大值对应索引return max_idx# 计算单个属性信息增益率
def c_entropy_gain_rate(data, label):ent = c_entropy(label)# 几个值循环几次count_value = Counter(data)sum_entropy = 0sum_iv = 0for value_a in count_value.keys():d = count_value[value_a] / len(label)sum_entropy += d * c_entropy(label[data == value_a])sum_iv -= d * np.log2(d)return (ent - sum_entropy) / sum_iv# 计算计算每个属性的信息增益率并求最大的
def find_entropy_ratio(data, label):# 返回大于中位数的下标def get_mid_index(list):mid = int(len(list)/2)b = sorted(enumerate(list), key=lambda x: x[1])  # x[1]是因为在enumerate(a)中，a数值在第1位c = [x[0] for x in b]  # 获取排序好后b坐标,下标在第0位return c[mid:]label = label.flatten()feature_count = data.shape[1]max_list_e = []max_list = []for i in range(feature_count):max_list_e.append(c_entropy_gain(data[:, i], label))max_list.append(c_entropy_gain_rate(data[:, i], label))# 只保留信息增益大于平均水平的max_list_mid = [max_list[i] for i in get_mid_index(max_list_e)]max_value = max(max_list_mid)  # 求列表最大值max_idx = max_list_mid.index(max_value)  # 求最大值对应索引return max_idx# 计算单个属性基尼指数
def c_gini(data, label):# 几个值循环几次count_value = Counter(data)gini = 0for value_a in count_value.keys():d = count_value[value_a] / len(label)class_gini = 1# 取出值一样的data的valuesame_value_class = label[data == value_a]count_labels = Counter(same_value_class)for i in count_labels.values():class_gini -= (i / count_value[value_a])**2gini += d * class_ginireturn gini# 计算计算每个属性的基尼指数并求最小的
def find_gini(data, label):label = label.flatten()feature_count = data.shape[1]min_list = []for i in range(feature_count):min_list.append(c_gini(data[:, i], label))min_value = min(min_list)  # 求列表最小值min_idx = min_list.index(min_value)  # 求最小值对应索引return min_idxdef sk(data_train, label_train, data_test, label_test):clf = tree.DecisionTreeClassifier(criterion="entropy")clf = clf.fit(data_train, label_train)score = clf.score(data_test, label_test)  # 返回精确度print("准确率为：", score * 100, "%")# 画决策树feature_name = ['色泽', '根蒂', '敲声', '纹理', '脐部', '触感']import matplotlib as mplmpl.rcParams['font.sans-serif'] = ['FangSong']  # 指定中文字体mpl.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题plt.rcParams['font.sans-serif'] = ['SimHei']plt.rcParams['axes.unicode_minus'] = False  # 正常显示负号fig, axes = plt.subplots(nrows=1, ncols=1, figsize=(4, 4), dpi=300)tree.plot_tree(clf,feature_names=feature_name,class_names=["好瓜", "坏瓜"],filled=True,  # 填充颜色，颜色越深，不纯度越低rounded=True  # 框的形状)plt.show()def pridect(data, mytree):# 找下个节点def find_result(mytree, row):# 如果是叶子节点if mytree.if_leaf:return mytree.label# 不是叶子节点就继续找row_the_value = row[mytree.node_label]return find_result(mytree.son[row_the_value], row)ret = np.array([find_result(mytree, row) for row in data])return retdef eveluate(predict, result):correct = 0for i in range(predict.shape[0]):correct += (int(predict[i][0]+0.5) == result[0])print("准确率为：", correct / predict.shape[0] * 100, "%")if __name__ == '__main__':random.seed(1129)data = pd.read_csv("watermelonData.csv").sample(frac=1, random_state=1129)# 因为西瓜数据集每列都是0到2的，所以这里就不进行标准化了labels = data["label"]data_shuffled = np.array(data[data.columns[:-1]])# 划分训练集测试集data_train = data_shuffled[:int(data_shuffled.shape[0]*0.6), :]label_train = np.array(labels[:data_train.shape[0]]).reshape(-1, 1)data_test = data_shuffled[data_train.shape[0]:, :]label_test = np.array(labels[data_train.shape[0]:]).reshape(-1, 1)choice = 0while choice != 5:print("1. ID3求解\n2. C45求解\n3. CART求解\n4. sklearn求解\n5. 退出")try:choice = int(input())except:breakif choice == 1:print("ID3求解中...")mytree = build_tree(data_train, label_train, choice)eveluate(pridect(data_test, mytree), label_test)# find_entropy_index(data_train, label_train)elif choice == 2:print("C45求解中...")mytree = build_tree(data_train, label_train, choice)eveluate(pridect(data_test, mytree), label_test)# find_entropy_ratio(data_train, label_train)elif choice == 3:print("CART求解中...")mytree = build_tree(data_train, label_train, choice)eveluate(pridect(data_test, mytree), label_test)# find_gini(data_train, label_train)elif choice == 4:print('sklearn yyds')sk(data_train, label_train, data_test, label_test)else:print("退出成功")break

参考：
吴恩达《机器学习》
sklearn官网
《百面机器学习》