pydotplus安装和Graphviz基本入门

最新推荐文章于 2025-06-01 23:20:35 发布

一页孤沙

最新推荐文章于 2025-06-01 23:20:35 发布

阅读量3.5k

点赞数 3

分类专栏： 4_数据挖掘与机器学习文章标签：决策树 pydotplus安装 Graphviz入门

4_数据挖掘与机器学习专栏收录该内容

28 篇文章

订阅专栏

博客介绍了利用Graphviz将Sklearn生成的dot格式决策树可视化的方法。强调安装顺序，先在cmd窗口安装pydotplus，再手动下载Graphviz的stable版本并添加到系统环境变量，最后进行Python3代码测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Graphviz是AT&T Labs Research开发的结构化图形绘制工具，支持多种格式输出。它的输入是一个用dot语言编写的绘图脚本，通过对输入脚本的解析，分析出其中的点，边以及子图，然后根据属性进行绘制。因此可利用Graphviz可以将Sklearn生成dot格式的决策树可视化。

一定要注意安装顺序

第一步、安装pydotplus

可以在 cmd 窗口中，直接使用下面pip或conda命令安装：
pip3 install pydotplus
conda install pydotplus

第二步、安装Graphviz

Graphviz 包不能使用pip或conda进行安装，否则安装完成后使用时报下面这种错误：
pydot.InvocationException: GraphViz’s executables not found：
手动安装可以解决这个问题，下载地址：http://www.graphviz.org/download/
选择下载stable版本

打开点击next默认安装，然后把 bin 默认安装路径C:\Program Files (x86)\Graphviz2.38\bin添加到系统环境变量PATH就可以使用了。

第三步Python3代码测试

from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus

iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

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 = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf('iris.pdf')

作者：nobodyyang
转载链接：https://www.jianshu.com/p/dd552f780a40