决策树可视化-Graphviz

graphviz

1.安装配置

• 下载graphviz-2.38.msi并安装。下载地址：https://graphviz.gitlab.io/_pages/Download/Download_windows.html
• 将安装路径"…\Graphviz\bin"添加至环境变量
• pip install graphviz

2.jupyter notebook中使用

import graphviz 
import os
from sklearn import tree

os.environ["PATH"] += os.pathsep + 'C:/Graphviz/bin' # 引入路径
dt = tree.DecisionTreeClassifier()
dt.fit(X=x.values,y=y.values)
dot_data = tree.export_graphviz(dt, out_file=None, 
                                feature_names = x.columns , class_names = True,
                                filled = True, rounded = True)
graph = graphviz.Source(dot_data)
graph