解决决策树可视化时python3.6没有PyGraphviz的问题

      最近在学习决策树,然后用python使决策树可视化时出了点问题。Graphviz,matplotlib的注解等都可以使决策树可视化。

      大多数版本是:

     1.安装GraphViz

      下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html,选择msi文件下载,安装即可。安装完成后,将bin目录加入到系统path环境变量中

    2.下载 PyGraphviz 并安装,

       pip 安装下载后的 whl 文件  

pip install pygraphviz-1.3.1-cp34-none-win_amd64.whl

   然后这时就出现了问题,因为python3.6版本没有pygraphviz ,如果你是2.7或者3.4的版本,可以安装这个。于是在网上又搜了搜,发现可以安装  pydotplus

       pip install pydotplus 即可安装

下面是简单的代码:

from sklearn.datasets import load_iris
from sklearn import tree
from IPython.display import  Image
import pydotplus

iris = load_iris()
X = iris.data[:, 2:]
y = iris.target

tree_clf = tree.DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)
dot_data = \
    tree.export_graphviz(
        tree_clf,
        out_file=None,
        feature_names=iris.feature_names[2:],
        class_names=iris.target_names,
        rounded=True,
        filled=True
)
graph = pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png()) #画出这个树
graph.write_png("tree1.png")

效果如下

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值