三种方法运行结果:python
方法四 决策树的可视化
#首先载入类库:
from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import os
#接着载入sciki-learn的自带数据,有决策树拟合,获得模型:
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
#如今能够将模型存入dot文件iris.dot。
with open("./iris.dot", 'w') as f:
f = tree.export_graphviz(clf,out_file=f)
#这时候咱们有3种可视化方法,第一种是用graphviz的dot命令生成决策树的可视化文件,
#敲完这个命令后当前目录就能够看到决策树的可视化文件iris.pdf.打开能够看到决策树的模型图。
#方法4.1 注意,这个命令在命令行执行
dot -Tpdf iris.dot -o iris.pdf
#方法4.2 用pydotplus生成iris.pdf。这样就不用再命令行去专门生成pdf文件了。
import pydotplus
dot_data = tree.export_graphviz(clf, out_file=None)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf("iris.dots.pdf")
os.environ['PATH'] += os.pathsep+ 'C:\Program Files (x86)\Graphviz2.38\bin'
#方法4.3 我的比较推荐的作法,由于这样能够直接把图产生在ipython的notebook。
from IPython.display import Image
import pydotplus
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)
Image(graph.create_png())
方法四结果:bash
欢迎关注小婷儿的博客:微信
csdn:https://blog.csdn.net/u010986753app
博客园:http://www.cnblogs.com/xxtalhr/post
有问题请在博客下留言或加做者微信:tinghai87605025 或 QQ :87605025学习
python QQ交流群:py_data 483766429spa
OCP培训说明链接:https://mp.weixin.qq.com/s/2cymJ4xiBPtTaHu16HkiuA.net
OCM培训说明链接:https://mp.weixin.qq.com/s/7-R6Cz8RcJKduVv6YlAxJA
小婷儿的python正在成长中,其中还有不少不足之处,随着学习和工做的深刻,会对以往的博客内容逐步改进和完善哒。
重要的事多作几遍。。。。。。