码迷,mamicode.com
首页 > 编程语言 > 详细

GraphViz和python交互

时间:2017-12-31 23:33:35      阅读:550      评论:0      收藏:0      [点我收藏+]

标签:out   imp   路径   方法   tree   编辑   进入   python   window   

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交互

标签:out   imp   路径   方法   tree   编辑   进入   python   window   

原文地址:https://www.cnblogs.com/oneTOinf/p/8159186.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!