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())
 机器学习-利用决策树解决多分类问题 随笔

 

扫码关注我们
微信号：SRE实战
拒绝背锅 运筹帷幄

SRE实战 互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。