sklearn 决策树可视化

您可以使用scikit-learn库中的可视化工具来可视化决策树模型。自版本0.21以后，scikit-learn为我们提供了一个非常方便的函数sklearn.tree.plot_tree()来实现这个目的。这个函数可以直接绘制决策树的图形。

另外，您还可以使用sklearn.tree.export_text()函数来将决策树以文本形式导出。这个函数可以将决策树的每个节点的判断条件以及叶子节点的类别输出为文本。您可以通过设置feature_names参数来指定特征的名称。

如果您已经训练好了一个决策树模型clf，并且想要将其可视化，可以使用下面的代码：

import matplotlib.pyplot as plt
from sklearn import tree

def plot_tree(clf):
    fig = plt.figure()
    tree.plot_tree(clf)
    fig.savefig("tree.png")

其中，clf是已经训练好的决策树模型，函数会将绘制的决策树保存为一个名为tree.png的图像文件。