1. 环境

  1. 纯python,3.6.8
  2. Windows10 x64系统

2. 安装Graphviz

2.1 安装软件

进入官方网站下载http://www.graphviz.org/download/,找到windows稳定版本的安装包
在这里插入图片描述
点进去后一路选择,我选择的是cmake的Release 2.44.1版本

在这里插入图片描述

安装的时候,记得选中将安装路径添加到path中。

在这里插入图片描述
打开cmd,运行dot -v,如果显示版本则说明安装成功。

在这里插入图片描述

2.2 安装库

参考https://www.mlln.cn/2018/09/25/python-graphviz%E5%9B%BE%E6%95%B0%E6%8D%AE%E5%8F%AF%E8%A7%86%E5%8C%96%E5%85%A5%E9%97%A8/#undefined

graphviz为Graphviz图形绘制软件提供了一个简单的纯Python接口。要使用pip安装它,请运行以下命令:

pip install graphviz

2.3 最重要的一步!!!

参考https://blog.csdn.net/qq_42479987/article/details/109403618

在命令行输入

dot -c

在命令行再输入

dot -v

如果出现以下字段,则说明能够使用Graphviz进行可视化了
在这里插入图片描述

3. 运行代码,进行可视化

参考https://sklearn.apachecn.org/docs/master/11.html
这里就是将文档中的代码一行行粘贴了过来,完整代码如下

>>> from sklearn import tree
>>> X = [[0, 0], [1, 1]]
>>> Y = [0, 1]
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(X, Y)
>>> clf.predict([[2., 2.]])
>>> clf.predict_proba([[2., 2.]])
>>> from sklearn.datasets import load_iris
>>> from sklearn import tree
>>> iris = load_iris()
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(iris.data, iris.target)
>>> import graphviz
>>> dot_data = tree.export_graphviz(clf, out_file=None)
>>> graph = graphviz.Source(dot_data)
>>> graph.render("iris")
>>> dot_data = tree.export_graphviz(clf, out_file=None,
...                      feature_names=iris.feature_names,  
...                      class_names=iris.target_names,  
...                      filled=True, rounded=True,  
...                      special_characters=True)  
>>> graph = graphviz.Source(dot_data)  
>>> graph

如果按照上面一步步来的话,最终应该能够顺利出图。
在这里插入图片描述

Logo

永洪科技,致力于打造全球领先的数据技术厂商,具备从数据应用方案咨询、BI、AIGC智能分析、数字孪生、数据资产、数据治理、数据实施的端到端大数据价值服务能力。

更多推荐