sklearn决策树可视化

sklearn决策树可视化需要借助Graphviz工具，具体步骤如下：

1. 安装Graphviz工具

首先需要在本地安装Graphviz工具，可以通过官网下载对应操作系统版本的安装包，并进行安装。

2. 安装Python库

使用以下命令安装相关Python库：

pip install pydotplus
pip install graphviz

3. 加载数据集并训练决策树模型

使用sklearn库加载数据集并训练决策树模型，例如：

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier

# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target

# 训练决策树模型
clf = DecisionTreeClassifier()
clf.fit(X, y)

4. 可视化决策树

使用以下代码将决策树可视化：

from sklearn.tree import export_graphviz
import pydotplus
from IPython.display import Image

# 将决策树导出为DOT格式
dot_data = export_graphviz(clf, out_file=None, 
                           feature_names=iris.feature_names,  
                           class_names=iris.target_names,  
                           filled=True, rounded=True,  
                           special_characters=True)

# 使用pydotplus库将DOT格式转换为图像
graph = pydotplus.graph_from_dot_data(dot_data)

# 使用Image库显示图像
Image(graph.create_png())

如果想将决策树保存为PDF或PNG文件，可以使用以下代码：

# 将决策树保存为PDF文件
graph.write_pdf("iris.pdf")

# 将决策树保存为PNG文件
graph.write_png("iris.png")

这样就可以将sklearn决策树可视化了。