运用graphviz进行决策树可视化中遇到的问题及解决方法

win10下graphziv的安装

graphziv下载

下载地址如下：
http://www.graphviz.org/download/
根据自己的系统选择相应的安装包

graphziv安装和环境变量的配置

这里勾选第二个选项，之后就不用手动进行环境变量的配置

查看是否安装成功

在命令窗口输入如下命令
dot -version

可能出现上述图片的错误，解决方法如下
输入如下命令：dot -c
然后继续输入命令：dot -version

这时候graphziv的版本和其他信息就表明graphziv安装成功

决策树可视化

anaconda下 pydotplus的安装

安装pydotplus之前确保graphziv安装成功
打开anaconda prompt，输入如下命令：
pip install pydotplus

决策树可视化例子

import pandas as pd
import matplotlib.pyplot as plt

# 加载数据
from sklearn.datasets.california_housing import fetch_california_housing
housing = fetch_california_housing()

# 构造决策树
from sklearn import tree
dtr = tree.DecisionTreeRegressor(max_depth=2)
dtr.fit(housing.data, housing.target)

# 决策树可视化
dot_data = tree.export_graphviz(
            dtr,
            out_file=None,
            feature_names=housing.feature_names,
            filled=True,
            impurity=False,
            rounded=True,
            special_characters=True)
import pydotplus
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf("dtr_white_background.pdf")

效果如下