关闭

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

标签: 机器学习
752人阅读 评论(0) 收藏 举报
分类:

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

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:58421次
    • 积分:2147
    • 等级:
    • 排名:第17632名
    • 原创:143篇
    • 转载:23篇
    • 译文:0篇
    • 评论:1条
    最新评论