機器學習-利用決策樹解決多分類問題

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())