1. 电脑安装graphviz
官网下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
下载好以后,win+R打开cmd命令窗口,定位到下载的msi文件位置:
按照程序顺序安装即可,默认安装在C盘。
然后到电脑中的:控制面板>系统和安全>系统>高级系统设置>环境变量中添加path,如下图:
1.win+R
2.输入命令:dot -version
3.观察到如下信息,则该设置生效;
此时再利用pip进行安装即可:
pip install 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")
1. 在命令行将路径转到保存iris.dot的文件路径中,使用cd命令。
2. 使用
dot -Tpng iris.dot -o iris.png
命令将 iris.dot文件转为iris.png图片。
查看图片,得到决策树模型图。
本文地址:https://blog.csdn.net/Ariel_jin/article/details/107171528
如对本文有疑问, 点击进行留言回复!!
听课笔记--Python数据分析--Numpy基础及基本应用
Python-定时任务APScheduler中两种调度器的区别
网友评论