windows10：
1，先要pip安裝pydotplus和graphviz：
pip install pydotplus pip install graphviz
2，www.graphviz.org下載msi文件并安裝。
3，系統環境變量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輸出決策樹圖形的例子就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
