三種方法運行結果: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正在成長中,其中還有不少不足之處,隨着學習和工做的深刻,會對以往的博客內容逐步改進和完善噠。
重要的事多作幾遍。。。。。。