sklearn.tree.plot_tree show 返回文本块而不是可视化树

Question

我试图使用plot_tree 显示树可视化，但它显示了一大块文本：

from sklearn.tree import plot_tree
plot_tree(t)

（其中 t 是 的实例DecisionTreeClassifier）这是输出：

[Text(464.99999999999994, 831.6, 'X[3] <= 0.8\nentropy = 1.581\nsamples = 120\nvalue = [39, 37, 44]'),
 Text(393.46153846153845, 646.8, 'entropy = 0.0\nsamples = 39\nvalue = [39, 0, 0]'),

等等等等。我如何让它显示视觉树？我使用的是 Jupyter 6.4.1，并且我已经在代码的前面导入了 matplotlib。谢谢！

Answer 1

就我而言，它适用于一个简单的“显示”：

plot_tree(t)
plt.show()