鸢尾花决策树可视化图片表示

from sklearn.datasets import load_iris # 导入类库
from IPython.display import Image
from sklearn import tree
import pydotplus
import graphviz
import sys

import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'
                                       # 注意后面的路径是你自己的graphviz的bin目录。
    
iris = load_iris()                     # 载入sciki-learn的自带数据
features = iris.data
target = iris.target

clf = tree.DecisionTreeClassifier()    # 载入决策树分类模型
clf = clf.fit(features, target)  # 决策树拟合，得到模型


dot_data = tree.export_graphviz(clf, 
out_file=None, 
feature_names=iris.feature_names, 
class_names=iris.target_names)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_png('iris.png')
Image(graph.create_png())