在机器学习的过程中，我们常常会用到树模型的方式来解决我们的问题。在工业界，我们不仅要针对某个问题利用机器学习的方法来解决问题，而且还需要能力解释其中的原理或原因。今天主要在这里记录一下树模型是怎么做可视化的方法：
1、首选需要用到几个包，需要导入一下。没有对应包的需要手动安装一下。
from IPython.display import Image from sklearn import tree import pydotplus
2、window下需要安装graphviz-2.38.msi。
安装的时候采用默认安装即可
3、执行如下代码即可展示可视化
import os os.environ["PATH"] += os.pathsep + ‘C:/Program Files (x86)/Graphviz2.38/bin/‘ dot_data = tree.export_graphviz(elem, out_file=None, #elem 是对应分类器 feature_names=select_featrue, #对应的特征 class_names=[‘0‘,‘1‘], #对应类别 filled=True, rounded=True, special_characters=True) graph = pydotplus.graph_from_dot_data(dot_data) graph.write_png(‘example.png‘) #保存图像 Image(graph.create_png())
原文：http://www.cnblogs.com/ljy2013/p/7687965.html