如何使用 Matplotlib 渲染决策树的完整图像

Question

我正在研究一个程序来创建决策树。下面的以下方法应保存树的图像：

def save_image(clf_, filename, feat_names, class_names, show=True):
    """Plot the tree and save to file."""
    plt.figure(dpi=200)
    tree.plot_tree(
        clf_, filled=True,
        feature_names=feat_names,
        class_names=class_names
    )
    plt.savefig(f'{filename}.png')
    if show:
        plt.show()

预期：

完整树的输出图像

实际：

输出的图像不是预期的。我该如何解决这个问题？