分类
DecisionTreeClassifier可以用于二分类和多分类。
from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.externals.six import StringIO
import pydot
iris=load_iris()
clf=tree.DecisionTreeClassifier()
clf=clf.fit(iris.data,iris.target)
//生成决策树图
dot_data=StringIO()
tree.export_graphviz(clf,out_file=dot_data,feature_names=iris.feature_names)
graph=pydot.graph_from_dot_data(dot_data.getvalue())
graph.write_png(‘tree.png’)