决策树的可视化输出

前面的博客里有提到决策树，我们也了解了决策树的构建过程，如果可以可视化决策树，把决策树打印出来，对我们理解决策树的构建会有很大的帮助。这篇文章中，我们就来看下如何可视化输出一棵决策树。

一、安装相应的插件

我们需要安装Graphviz和pygraphviz，教程，这个教程里有详细的安装过程，这里就不赘述了。

二、运行实例

我们依然以iris数据集为例，打印输出决策树看下效果。

#coding=utf-8
import  numpy as np
import pandas as  pd
import pydotplus
def lrisTrain():
  #预处理-引入鸢尾数据：
  from  sklearn.datasets import load_iris
  iris = load_iris()

  from sklearn.cross_validation import train_test_split
  # 把数据分为测试数据和验证数据
  trai