Sklearn plot_tree图太小

Question

我有这个简单的代码：

clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, y)

tree.plot_tree(clf.fit(X, y))
plt.show()

我得到的结果是这张图：

如何使该图清晰可见？我正在使用PyCharm Professional 2019.3作为我的IDE。

Answer 1

我认为您要查找的设置是fontsize。您必须将其与max_depth和figsize进行平衡才能获得清晰的图。这是一个例子

from sklearn import tree
from sklearn.datasets import load_iris
import matplotlib.pyplot as plt

X, Y = load_iris(return_X_y=True)
clf = tree.DecisionTreeClassifier()
fig, ax = plt.subplots(figsize=(12, 12))
tree.plot_tree(clf.fit(X, Y), max_depth=4, fontsize=10)
plt.show()

如果要捕获整个树的结构，我想以小字体和高dpi保存图是解决方案。然后，您可以打开图片并缩放到特定节点以进行检查。

clf = tree.DecisionTreeClassifier()
fig, ax = plt.subplots(figsize=(50, 24))
tree.plot_tree(clf.fit(X, Y), fontsize=6)
plt.savefig('tree_high_dpi', dpi=100)

这是在大树上的样子的一个例子。

Answer 2

如何设置手前图像的大小：

receiver