码迷,mamicode.com
首页 > 编程语言 > 详细

sklearn中决策树算法DesiciontTreeClassifier()调用以及sklearn自带的数据包sklearn.datasets.load_iris()的应用

时间:2018-06-06 12:21:48      阅读:418      评论:0      收藏:0      [点我收藏+]

标签:sed   rate   code   lin   数据   dict   sci   splay   hid   

决策树方法的简单调用记录一下

技术分享图片
 1 clf=tree.DecisionTreeClassifier()
 2 dataMat=[];labelMat=[]
 3 dataPath=D:/machinelearning data/machinelearninginaction/Ch05/testSet.txt
 4 fr = open(dataPath)
 5 for line in fr.readlines():  # readilnes()将文件内容存在列表里
 6     lineArr = line.strip().split()  # 去掉空格
 7     labelMat.append(int(lineArr[-1]))
 8     dataMat.append([float(lineArr[0]), float(lineArr[1])])  
 9 x=np.array(dataMat)
10 y=np.array(labelMat)
11 clf.fit(x,y)
12 yHat=clf.predict(x)
13 result=np.count_nonzero(yHat==y)
14 print(正确个数,result)
15 print(正确率,result/len(yHat))
View Code

并附上介绍决策树的链接http://scikit-learn.org/stable/modules/tree.html

sklearn中自带的数据应用sklearn.datasets.load_iris()的调用以及相应的应用链接:http://scikit-learn.org/stable/modules/generated/sklearn.datasets.load_iris.html#sklearn.datasets.load_iris

sklearn中决策树算法DesiciontTreeClassifier()调用以及sklearn自带的数据包sklearn.datasets.load_iris()的应用

标签:sed   rate   code   lin   数据   dict   sci   splay   hid   

原文地址:https://www.cnblogs.com/zz22--/p/9143888.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!