决策树(Decision Trees ,DTs)是一种无监督的学习方法,用于分类和回归。它对数据中蕴含的决策规则建模,以预测目标变量的值。
某些情况,例如下面的例子,决策树通过学习模拟一个包含一系列是否判断的正弦曲线。树越深,决策树的规则和拟合越复杂。
(图)
决策树的一些优点:
决策树的一些缺点:
DecisionTreeClassifier 能够对数据进行多分类的类。
和其他分类器一样,DecisionTreeClassifier 有两个向量输入:X,稀疏或密集,大小为[n_sample,n_fearure],存放训练样本; Y,值为整型,大小为[n_sample],存放训练样本的分类标签:
>>> 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,1,K-1])。
用数据集 lris ,可以构造下面的树:
>>> 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 数据集训练树的例子。
>>> from sklearn.externals.six import StringIO >>> with open("iris.dot", ‘w‘) as f: ... f = tree.export_graphviz(clf, out_file=f)
啊啊啊啊
原文:http://www.cnblogs.com/zhaoxy/p/5054938.html