您好,登錄后才能下訂單哦!
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 #確認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)容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。