1,工具和平台:
python2.7 windows
2,决策树的可视化展示据我所知有两种途径:一是将生成的结果导出为pmml文件,工具包为sklearn2pmml等,具体可见https://github.com/jpmml/jpmml-sklearn 和https://github.com/jpmml/sklearn2pmml 如果遇到安装问题可以留言;而是进行graphviz的图形化展示,首先需要安装graphviz-2.38.msi url:http://www.graphviz.org/pub/graphviz/stable/windows/graphviz-2.38.msi,
这是因为python所用的工具包pydot依赖于dot.exe,配置
path:C:\Program Files (x86)\Graphviz2.38\bin 和 C:\Program Files (x86)\Graphviz2.38 具体按照你自己的安装目录即可
接着安装pydot : pip install pydot
基本原理为:利用sklearn.tree.export_graphviz 可以将随机森林中每一棵树的结构以doc的形式表示出来,结合graphviz可以将结构图转为png或者pdf等。
#coding=utf-8
from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
from sklearn.externals.six import StringIO
import pydot
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph[0].write_dot('iris_simple.dot')
graph[0].write_png('iris_simple.png')
可能会出现错误如: