1. 程式人生 > >GraphViz和python交互

GraphViz和python交互

spa port sklearn 模型 學習 方法 tree out rom

graphviz強大而便捷的關系圖/流程圖繪制方法,很容易讓我們聯想到機器學習中的Decision Tree的展示方式。幸運的是,scikit-learn提供了生成.dot文件的接口,具體操作如下:

python編輯環境下:

from sklearn.tree import export_graphviz    # 導入的是一個函數
# tree表示已經訓練好的模型,即已經調用過DecisionTreeClassifier實例的fit(X_train, y_train)方法
export_graphviz(tree, out_file='tree.dot', 
        feature_names=
['petal length', 'petal width'])1234

進入windows命令行界面,cd 切換到tree.dot所在的路徑,執行

dot -Tpng tree.dot -o tree.png

技術分享圖片

GraphViz和python交互