提问者:小点点

SkLearning决策树plot_tree节点是重叠的


当我使用sklearn.tree.plot_tree()绘制skLearning决策树时,节点在更深的层次上重叠,我无法读取节点中的内容。展示你的结果是不好的。它看起来像这样:

我如何在它们之间获得更多的空间,以避免重叠?

我已经找到了这个问题,我已经将fontsize增加到10,figsize增加到(20,30),但是它导致了如图所示绘制的树。我还将新闻部增加到150。这是我的代码:

fig = plt.figure(figsize=(20,30))
artists = sklearn.tree.plot_tree(decision_tree=clf, feature_names=feature_names, class_names=class_names, filled=True, rounded=True, fontsize=10, max_depth=4)
fig.savefig(filepath, dpi=150)

共1个答案

匿名用户

您需要控制渲染图像的大小。尝试改变像fig=plt.figure(figsize=(50,30))这样的数字。
你可以在https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html.找到原来的留档