python輸出決策樹圖形的例子
更新時間:2019年08月09日 14:45:35 作者:CrazyTTT
今天小編就為大家分享一篇python輸出決策樹圖形的例子,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
windows10:
1,先要pip安裝pydotplus和graphviz:
pip install pydotplus pip install graphviz
2,www.graphviz.org下載msi文件并安裝。
3,系統(tǒng)環(huán)境變量path中增加兩項:
C:\Program Files (x86)\Graphviz2.38\bin C:\Program Files (x86)\Graphviz2.38 #確認(rèn)graphviz是安裝在上面路徑當(dāng)中。
4,python中使用方法:
from sklearn.externals.six import StringIO
import pydotplus
#drt是DecisionTreeClassifier(),在之前要fit訓(xùn)練之后才能在這里輸出圖形。
dot_data = StringIO()
tree.export_graphviz(drt, out_file=dot_data)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_png("out.png") #當(dāng)前文件夾生成out.png
#這三行代碼可以生成pdf:
dot_data = tree.export_graphviz(drt, out_file=None)
graph = graphviz.Source(dot_data)
graph.render()
以上這篇python輸出決策樹圖形的例子就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python spilt()分隔字符串的實現(xiàn)示例
split() 方法可以實現(xiàn)將一個字符串按照指定的分隔符切分成多個子串,本文介紹了spilt的具體使用,感興趣的可以了解一下2021-05-05
python解析Chrome瀏覽器歷史瀏覽記錄和收藏夾數(shù)據(jù)
大家好,本篇文章主要講的是python解析Chrome瀏覽器歷史瀏覽記錄和收藏夾數(shù)據(jù),感興趣的同學(xué)趕快來看一看吧,對你有幫助的話記得收藏一下2022-02-02
Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)找出序列中出現(xiàn)次數(shù)最多的元素算法示例
這篇文章主要介紹了Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)找出序列中出現(xiàn)次數(shù)最多的元素算法,涉及Python collections模塊中的Counter類相關(guān)使用技巧與操作注意事項,需要的朋友可以參考下2018-03-03
解決tensorboard多個events文件顯示紊亂的問題
今天小編就為大家分享一篇解決tensorboard多個events文件顯示紊亂的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02

