决策树图形化

2018/10/24

最开始学机器学习的时候，看的一本书《机器学习系统设计》（可能是，记不清了， 反正是最开始学的几个书）里面有个章节介绍决策树的时候，他就说到了这个画图的部分，但是他的呢，是自己画的，我记得。
不过sklearn提供了这样的途径，转载自：http://scikit-learn.org/stable/modules/tree.html
这个里面就说明了这个事情。
但是要安装一些库。
pip install graphviz
这个实现就已经安装好了，不知道是什么时候安装好的。
但他还需要graphviz的引擎。
一开始以为要去官网下载，下了rpm之后，发现很多依赖库都没安装，所以就先源码安装。
不过，用yum尝试了一下，发现yum源里有合格东西。
yum install graphviz
这样就安装好了。

---

通过看其官网介绍的内容，

>>> import graphviz 
>>> dot_data = tree.export_graphviz(clf, out_file=None) 
>>> graph = graphviz.Source(dot_data) 
>>> graph.render("iris")

我在自己代码里面也是这么执行的，当然名字自然是改了。

这样执行的结果是，他会生成两个文件，一个就是我要的那个图片。可以看一下自己做出来的效果。

决策树模型

这个图片他生出的是一个pdf文件。

另外一个文件是无后缀的，这个文件的内容就是他内部的一些数据结构，用来代表这个模型的，主要是一些参数。（这个文件后续如果有想法的画，可以做一下C源码的移植，这个还是挺有意义的）

---

上面这个图呢，画出来的是最简单的。看了他们弄的那个，还是很漂亮的。所以，后续如果需要的话，可以做一下。