無法使用 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')