决策树可视化

决策树可视化

过去，关于sklearn决策树可视化的教程大部分都是基于Graphviz（一个图形可视化软件）的。

Graphviz的安装比较麻烦，并不是通过pip install就能搞定的，因为要安装底层的依赖库。

现在，自版本0.21以后，scikit-learn也自带可视化工具了，它就是sklearn.tree.plot_tree()

假设决策树模型（clf)已经训练好了，画图的代码如下：

def tree1(clf):
    fig = plt.figure()
    tree.plot_tree(clf)
    fig.savefig(os.path.join(fig_dir, "tree1.png"))

下面figsize为画布大小，fontsize为字体大小。

def tree(clf):
    fig = plt.figure(figsize=(35, 10))
    tree.plot_tree(clf, fontsize=8)
    fig.savefig(os.path.join(fig_dir, "tree.png"))

转载自决策树可视乎