Python机器学习算法库scikit-learn学习之决策树实现方法详解

更新时间：2020-07-11 17:30:01 作者：startmvc

本文实例讲述了Python机器学习算法库scikit-learn学习之决策树实现方法。分享给大家供大家参

本文实例讲述了Python机器学习算法库scikit-learn学习之决策树实现方法。分享给大家供大家参考，具体如下：

决策树

决策树(DTs)是一种用于分类和回归的非参数监督学习方法。目标是创建一个模型，通过从数据特性中推导出简单的决策规则来预测目标变量的值。

例如，在下面的例子中，决策树通过一组if-then-else决策规则从数据中学习到近似正弦曲线的情况。树越深，决策规则越复杂，模型也越合适。

这里写图片描述

决策树的一些优势是:

便于说明和理解，树可以可视化表达；
需要很少的数据准备。其他技术通常需要数据标准化，需要创建虚拟变量，并删除空白值。
使用树的成本是用于对树进行训练的数据点的对数。
能够处理数值和分类数据。其他技术通常是专门分析只有一种变量的数据集。
能够处理多输出问题。
使用白盒模型。如果一个给定的情况在模型中可以观察到，那么这个条件的解释很容易用布尔逻辑来解释。相比之下，在黑盒模型中(例如:在人工神经网络中，结果可能更难解释。
可以使用统计测试验证模型。这样就可以解释模型的可靠性。
即使它的假设在某种程度上违反了生成数据的真实模型，也会表现得很好。

决策树的缺点包括:

决策树学习可能会生成过于复杂的树，不能代表普遍的规则，这就称为过拟合。修剪(目前不支持的)机制，设置叶片节点所需的最小样本数目或设置树的最大深度是避免此问题的必要条件。
决策树可能不稳定，因为数据中的小变化可能导致生成完全不同的树。这个问题通过在一个集合中使用多个决策树来减轻。
实际的决策树学习算法是基于启发式算法的，例如在每个节点上进行局部最优决策的贪婪算法。这种算法不能保证返回全局最优决策树。通过在集合学习者中训练多个树，可以减少这种情况，在这里，特征和样本是随机抽取的。
有些概念很难学习，因为决策树无法很容易地表达它们，例如XOR、奇偶性或多路复用问题。

分类

决策树分类器（DecisionTreeClassifier）是一个能够在数据集上执行多类分类的类。

与其他分类器一样，决策树分类器以输入两个数组作为输入:数组X，稀疏或密集，[n_samples,n_features]保存训练样本，以及数组Y的整数值，[n_samples]，保存训练样本的类标签:


>>> from sklearn import tree
>>> X = [[0, 0], [1, 1]]
>>> Y = [0, 1]
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(X, Y)

模型可以用来预测样本的类别:


>>> clf.predict([[2., 2.]])
array([1])

或者，可以预测每个类的概率，在叶片上同一类的训练样本的分数


>>> clf.predict_proba([[2., 2.]])
array([[ 0., 1.]])

DecisionTreeClassifier可以同时进行二进制(其中标签为[- 1,1])分类和多类(标签为[0]，……,k - 1])分类。

使用虹膜数据集，我们可以构建如下的树:


>>> from sklearn.datasets import load_iris
>>> from sklearn import tree
>>> iris = load_iris()
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(iris.data, iris.target)

训练之后，我们可以使用export_graphviz 将树导出为Graphviz格式。下面是一个在整个iris（虹膜）数据集上输出的树示例:


>>> with open("iris.dot", 'w') as f:
... f = tree.export_graphviz(clf, out_file=f)

然后我们可以使用Graphviz的dot工具来创建一个PDF文件(或者任何其他受支持的文件类型):dot -Tpdf iris.dot -o iris.pdf


>>> import os
>>> os.unlink('iris.dot')

或者，如果我们安装了Python模块pydotplus，我们可以在Python中直接生成PDF文件(或任何其他受支持的文件类型):


>>> import pydotplus
>>> dot_data = tree.export_graphviz(clf, out_file=None)
>>> graph = pydotplus.graph_from_dot_data(dot_data)
>>> graph.write_pdf("iris.pdf")

export_graphviz exporter 还支持各种各样的选项，包括根据它们的类(或用于回归的值)着色节点，如果需要的话，还可以使用显式变量和类名IPython还可以使用Image()函数来显示这些情节:


>>> from IPython.display import Image
>>> dot_data = tree.export_graphviz(clf, out_file=None,
 feature_names=iris.feature_names,
 class_names=iris.target_names,
 filled=True, rounded=True,
 special_characters=True)
>>> graph = pydotplus.graph_from_dot_data(dot_data)
>>> Image(graph.create_png())

这里写图片描述

模型可以用来预测样本的类别:


>>> clf.predict(iris.data[:1, :])
array([0])

或者，可以预测每个类的概率，这是同一类在叶子中的训练样本的分数:


>>> clf.predict_proba(iris.data[:1, :])
array([[ 1., 0., 0.]])

这里写图片描述

Examples:

Plot the decision surface of a decision tree on the iris dataset

回归

这里写图片描述

使用决策树类回归，决策树也可以应用于回归问题。

在分类设置中，fit方法将数组X和y作为参数，只有在这种情况下，y被期望有浮点值而不是整数值:


>>> from sklearn import tree
>>> X = [[0, 0], [2, 2]]
>>> y = [0.5, 2.5]
>>> clf = tree.DecisionTreeRegressor()
>>> clf = clf.fit(X, y)
>>> clf.predict([[1, 1]])
array([ 0.5])

Examples:

Decision Tree Regression

多输出问题

一个多输出问题是一个受监督的学习问题，有几个输出可以预测，即当Y是一个二维数组[n_samples,n_output]。

当输出之间没有相关性时，解决这类问题的一个非常简单的方法是建立n个独立的模型，即每一个输出，然后使用这些模型独立地预测每一个输出。

然而，因为可能与相同输入相关的输出值本身是相关的，通常更好的方法是构建一个能够同时预测所有n输出的单一模型。首先，它需要较低的培训时间，因为只构建了一个估计值。其次，结果估计量的泛化精度通常会增加。

对于决策树，这种策略可以很容易地用于支持多输出问题。这需要以下更改: