sklearn决策树的结构

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

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

Thx!


 

山中人   2017-10-05 13:08



   1个回答 
3

需要通过调用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里的RandomForestClassifier怎么选样本的数量   1回答

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

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

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

python画散点图   1回答

用plt.hist画直方图,怎么返回每个bins的上下界和个数?   1回答

matplotlib中bar plot横坐标的数字改成自定义的文字   1回答

如何用python画饼图?   1回答

seaborn如何显示图?   2回答

plt.pie画饼图,怎么调节透明度alpha   1回答

plt.bar怎么设置每个bar的颜色?   1回答



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

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

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

我们谢绝答非所问。

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

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