欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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
#確認graphviz是安裝在上面路徑當中。

4,python中使用方法:

from sklearn.externals.six import StringIO 
import pydotplus

#drt是DecisionTreeClassifier(),在之前要fit訓練之后才能在這里輸出圖形。
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") #當前文件夾生成out.png

#這三行代碼可以生成pdf:
dot_data = tree.export_graphviz(drt, out_file=None) 
graph = graphviz.Source(dot_data) 
graph.render() 

以上這篇python輸出決策樹圖形的例子就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關文章

最新評論