无法使用 Matplotlib 保存决策树 plot

Question

字体太小而无法可视化，所以我希望保存图像并在本地而不是在 Jupyter 上查看。

fig, axes = plt.subplots(nrows = 1,ncols = 1,figsize = (4,4), dpi=300)
tree.plot_tree(clf,filled=True,rounded=True)
plt.savefig('foo.png')

但是，保存的图像完全是空白的。

我怎样才能解决这个问题？

Answer 1

尝试在斧头上绘图：

fig, axes = plt.subplots(nrows = 1,ncols = 1,figsize = (4,4), dpi=300)
tree.plot_tree(clf,filled=True,rounded=True, ax=axes)
plt.savefig('foo.png')