安裝GraphViz
下載地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
選擇msi文件下載,安裝即可。安裝完成后,將bin目錄加入到系統path環境變量中
安裝graphviz的python庫
使用conda install python-graphviz
命令安裝即可
生成可視化文件
import graphviz # doctest: +SKIP
from sklearn import tree
print(data.columns)
#ID3為決策樹分類器fit之后得到的模型,注意這里必須在fit后執行,在predict之后運行會報錯
dot_data = tree.export_graphviz(ID3, out_file=None,feature_names=data.columns[:-1],class_names=np.unique(y)) # doctest: +SKIP
graph = graphviz.Source(dot_data) # doctest: +SKIP
#在同級目錄下生成tree.pdf文件
graph.render("tree") # doctest: +SKIP
參考http://sklearn.apachecn.org/cn/0.19.0/modules/tree.html#tree-algorithms