參考這篇文章即可。
http://m.blog.csdn.net/lanchunhui/article/details/49472949
和python交互
graphviz強(qiáng)大而便捷的關(guān)系圖/流程圖繪制方法冠绢,很容易讓我們聯(lián)想到機(jī)器學(xué)習(xí)中的Decision Tree的展示方式崇呵。幸運(yùn)的是,scikit-learn提供了生成.dot文件的接口,具體操作如下:
在python編輯環(huán)境下:
1 from sklearn.tree importexport_graphviz# 導(dǎo)入的是一個(gè)函數(shù)# tree表示已經(jīng)訓(xùn)練好的模型,即已經(jīng)2 調(diào)用過DecisionTreeClassifier實(shí)例的fit(X_train, y_train)方法
3 export_graphviz(tree, out_file='tree.dot',? ? ? ?
4 ?feature_names=['petal length','petal width'])
進(jìn)入windows命令行界面呛每,cd 切換到tree.dot所在的路徑恤溶,執(zhí)行
dot-Tpng tree.dot-o tree.png