Scikit-Learn (sklearn) Python 包有一个很好的函数 sklearn.tree.plot_tree来绘制（决策）树。
默认情况下可以通过如下命令进行绘图
tree.plot_tree(clf)如果您尝试从像 Spyder 这样的 IDE 中保存它，它的分辨率可能会很低。
解决方法是先导入matplotlib.pyplot：
import matplotlib.pyplot as plt然后，以下代码将允许您将 sklearn 树保存为 .eps（或者您可以相应地更改格式）：
plt.figure()
tree.plot_tree(clf,filled=True)
plt.savefig('tree.eps',format='eps',bbox_inches = "tight")clf 是您的决策树分类器（在绘制树之前定义）：
# Example from https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html
clf = tree.DecisionTreeClassifier(random_state=0)
clf = clf.fit(iris.data, iris.target)输出结果是矢量图形格式 (.eps) 树，在放大时将保留其完整分辨率。 bbox_inches=”tight” 命令可防止截断图像。如果没有该命令，有时 sklearn 树将被裁掉并且不完整。
原创文章，作者：somirror，如若转载，请注明出处：https://somirror.com/5262.html
