参考这篇文章即可。
http://m.blog.csdn.net/lanchunhui/article/details/49472949
和python交互
graphviz强大而便捷的关系图/流程图绘制方法,很容易让我们联想到机器学习中的Decision Tree的展示方式。幸运的是,scikit-learn提供了生成.dot文件的接口,具体操作如下:
在python编辑环境下:
1 from sklearn.tree importexport_graphviz# 导入的是一个函数# tree表示已经训练好的模型,即已经2 调用过DecisionTreeClassifier实例的fit(X_train, y_train)方法
3 export_graphviz(tree, out_file='tree.dot',
4 feature_names=['petal length','petal width'])
进入windows命令行界面,cd 切换到tree.dot所在的路径,执行
dot-Tpng tree.dot-o tree.png