机器学习-利用决策树解决多分类问题

from sklearn.datasets import load_iris
import pydotplus
from IPython.display import Image
from sklearn import tree
#训练模型
iris=load_iris()
clf=tree.DecisionTreeClassifier()
clf=clf.fit(iris.data,iris.target)

#绘图
dot_data=tree.export_graphviz(decision_tree=clf,
                              out_file=None,
                             rounded=True,
                              filled=True,
                              feature_names=iris.feature_names)

graph=pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png())