决策树可视化

相应软件安装

pip install graphviz

cmake安装

下载地址

软件安装过程，要勾选添加环境变量：

然后一路下一步，就可以了~

graphviz安装

安装了Python库graphviz，还不够，还需要安装graphviz下载页面这个软件，下载地址如下：软件

软件安装过程，要勾选添加环境变量：

然后一路下一步，就可以了~

决策树画图实操

常规导出到图片（没有颜色设置）

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier 
iris = load_iris()
clf = DecisionTreeClassifier()
clf_iris = clf.fit(iris.data, iris.target)

import graphviz
# 将决策树的结构数据导出
dot_data = tree.export_graphviz(clf, out_file = './iris')
graph = graphviz.Source.from_file('./iris')
graph.render('iris',format='pdf')# 保存到文件中

# 不导出直接，使用
dot_data = tree.export_graphviz(clf, out_file = None)
graph = graphviz.Source(dot_data)
graph.render('iris',format='png')# 保存到文件中

以下是，保存结果显示：

pdf：

png：

更多设置导出到图片（彩色决策树可视化图片）

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier 
iris = load_iris()
clf = DecisionTreeClassifier()
clf_iris = clf.fit(iris.data, iris.target)

dot_data = tree.export_graphviz(clf, out_file=None, 
                            feature_names=iris.feature_names,# 特征名
                            class_names=iris.target_names,# 类别名
                            filled=True, # 填充颜色
                            rounded=True) # 圆角

graph = graphviz.Source(dot_data)
graph.render('iris2') # 默认保存pdf格式

以下是，保存结果显示效果：

---

我要做的事情是，庖丁解书，让你看个通透，学个明白。既有文章，又有视频。包教包会，学不会一定不是学生的问题，而是老师的问题！

代码点亮人生，思维改变世界~