sklearn决策树的结构

  统计/机器学习 监督式学习 数据可视化 Python    浏览次数: 540
4

我用sklearn构造了决策树模型后,可以有办法得到这个决策树的结构?也就是if-else的判断的规则。

Thx!


 

山中人   2017-10-05 13:08



   1个回答 
2

需要通过调用tree.export_graphviz这个方法。

以sklearn自带的数据集iris为例

from sklearn.datasets import load_iris
from sklearn import tree

clf = tree.DecisionTreeClassifier()
iris = load_iris()
clf = clf.fit(iris.data, iris.target)
tree.export_graphviz(clf, out_file='trees.txt')

输出的trees.txt文件里有保存了你的决策树的判断规则。


如果你想把这个规则画出来的话,可以借用这个网站http://webgraphviz.com/,把上面的txt文件里的内容,复制过去就行了,可以得到类似下面的图。



数据痴汉   2017-10-15 10:58



  相关主题

sklearn随机森林设置每棵树的权重   2回答

sklearn模块里的DecisionTreeClassifier为什么每次结果不同(数据相同的情况下)   1回答

对sklearn的决策树进行剪枝   2回答

sklearn里的RandomForestClassifier怎么选样本的数量   1回答

plt.scatter画二元分类散点图,怎么用加号表示阳性、减号表示阴性   1回答

python里怎么实现柱状图?   1回答

python里画出“热图”   3回答

matplotlib.pyplot中怎么把legend放到图片外面去?   1回答

matplotlib.pyplot环境下给曲线下方的面积涂色?   1回答

python做图自定义背景色   1回答

plt.scatter plot怎么让不同的类别有不同的颜色   2回答

matplotlib.pyplot画图怎么添加横轴,纵轴的名字?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!