用python绘制决策树

tquggr8v 于 2023-02-15 发布在 Python

关注(0)|答案(1)|浏览(130)

我在用python做画图决策树，

tree.plot_tree(clf.fit(X_train, y_train))
plt.suptitle("Decision surface of a decision tree using paired features")
plt.show()

但当我运行这段代码时树显示如下
决策树

有没有可能让树正常化？

1条答案

export_graphviz可能是一个不错的选择。
下面是一个Kaggle notebook，它解释了如何充分利用sklearn提供的这个功能。
对于plot_tree函数，你可以尝试使用fontsize参数来改变字体大小，或者，如果你还没有改变图形的大小，你也可以尝试这样做：

fig = matplotlib.pyplot.gcf()
fig.set_size_inches(x, y) #dimensions you want