python生成决策树图

使用Python可以使用Graphviz库来生成决策树图。以下是一个简单的示例代码：

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import graphviz

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

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

# 生成决策树图形
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)  
graph = graphviz.Source(dot_data)  
graph.render("iris") # 保存图形到文件 iris.pdf

该代码将使用决策树对鸢尾花数据集进行分类，并生成对应的决策树图形，保存为iris.pdf文件。你需要安装scikit-learn和Graphviz库才能运行该代码。