Python/scikit-learn机器学习库(决策树)

分类

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