首页 > 解决方案 > 如何在 Python 中显示决策树的高分辨率图片?

问题描述

我正在使用以下代码来提取规则。请帮助我绘制更高分辨率的树,因为当我增加树深度时图像会变得模糊。

from sklearn.tree import DecisionTreeClassifier
from sklearn import tree
classifier = DecisionTreeClassifier(max_depth = 3,random_state = 0)
tree.plot_tree(classifier);

另外,如何从随机森林分类器中提取规则。

也有

标签: pythonmachine-learningclassificationdecision-tree

解决方案


我只需将绘图保存到 PDF 文件并使用它放大到您想要的任何部分

只是plt.savefig('out.pdf')在你打电话之后plot_tree,matplotlib 应该做正确的事

请注意,任何“矢量”图像格式都可以在这里做正确的事情,因此svg文件也是一个不错的选择,因为图像编辑器倾向于更好地支持缩放


推荐阅读