首先,安装graphviz,分为三步
- 电脑安装graphviz安装包
- 设置环境变量
- 为Python加载Graphviz
1. 电脑安装graphviz
官网下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
下载好以后,win+R打开cmd命令窗口,定位到下载的msi文件位置:
按照程序顺序安装即可,默认安装在C盘。
然后到电脑中的:控制面板>系统和安全>系统>高级系统设置>环境变量中添加path,如下图:
测试安装:
1.win+R
2.输入命令:dot -version
3.观察到如下信息,则该设置生效;
加载graphviz
此时再利用pip进行安装即可:
pip install graphviz
graphviz导出模型
scikit-learn的Decision Trees (DTs) 范例
利用Iris数据集,我们可以构建如下树:
>>> from sklearn.datasets import load_iris >>> from sklearn import tree >>> X, y = load_iris(return_X_y=True) >>> clf = tree.DecisionTreeClassifier() >>> clf = clf.fit(X, y)
一旦训练完成,你可以用plot_tree函数绘制树图:
>>> tree.plot_tree(clf)
以下是在整个虹膜数据集上训练的上述树的示例graphviz导出;结果保存在输出文件中iris.dot
:
>>> dot_data = tree.export_graphviz(clf, out_file="iris.dot")
使用命令行将 iris.dot训练树模型转成png图片可视化
1. 在命令行将路径转到保存iris.dot的文件路径中,使用cd命令。
2. 使用
dot -Tpng iris.dot -o iris.png
命令将 iris.dot文件转为iris.png图片。
查看图片,得到决策树模型图。