sklearn 决策树画图

要使用sklearn绘制决策树图，需要安装graphviz和pydotplus两个库。安装完成后，可以按照以下步骤进行绘制决策树图。

1. 导入需要的库和数据集

from sklearn import tree
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
import graphviz
import pydotplus

iris = load_iris()
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.3, random_state=0)

2. 训练决策树模型

clf = tree.DecisionTreeClassifier()
clf.fit(X_train, y_train)

3. 绘制决策树图

dot_data = tree.export_graphviz(clf, out_file=None, 
                      feature_names=iris.feature_names,  
                      class_names=iris.target_names,  
                      filled=True, rounded=True,  
                      special_characters=True)  
graph = pydotplus.graph_from_dot_data(dot_data)  
graph.write_pdf("iris.pdf")

以上代码将生成一个名为“iris.pdf”的决策树图。可以使用Adobe Acrobat Reader或其他PDF阅读器打开该文件，查看决策树图。