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

python决策树

时间:2020-08-18 15:46:03      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:dig   rac   8 8   load   sklearn   抽取   创建ca   col   and   

 

一、CART算法的实现

#encoding:utf-8
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_digits
#准备数据
digit = load_digits()
data  = digit.data
target = digit.target
#随机抽取33%的数据做测试集,其余为训练集
train_data,test_data,train_target,test_target = train_test_split(data,target,test_size=0.33,random_state=0)
#创建CART分类树
clf = DecisionTreeClassifier(criterion=gini)
#拟合构造CART分类树
clf = clf.fit(train_data,train_target)
#用CART分类树做预测
test_predict = clf.predict(test_data)
#将结果输出
print(实际结果为:,test_target,--预测结果为:,test_predict)
#预测结果的准确率
score = accuracy_score(test_target,test_predict)
print("CART分类树准确率%.4f" % score)

结果:

实际结果为: [2 8 2 6 6 7 1 9 8 5 2 8 6 6 6 6 1 0 5 8 8 7 8 4 7 5 4 9 2 9 4 7 6 8 9 4 3
 1 0 1 8 6 7 7 1 0 7 6 2 1 9 6 7 9 0 0 5 1 6 3 0 2 3 4 1 9 2 6 9 1 8 3 5 1
 2 8 2 2 9 7 2 3 6 0 5 3 7 5 1 2 9 9 3 1 7 7 4 8 5 8 5 5 2 5 9 0 7 1 4 7 3
 4 8 9 7 9 8 2 6 5 2 5 8 4 8 7 0 6 1 5 9 9 9 5 9 9 5 7 5 6 2 8 6 9 6 1 5 1
 5 9 9 1 5 3 6 1 8 9 8 7 6 7 6 5 6 0 8 8 9 8 6 1 0 4 1 6 3 8 6 7 4 5 6 3 0
 3 3 3 0 7 7 5 7 8 0 7 8 9 6 4 5 0 1 4 6 4 3 3 0 9 5 9 2 1 4 2 1 6 8 9 2 4
 9 3 7 6 2 3 3 1 6 9 3 6 3 2 2 0 7 6 1 1 9 7 2 7 8 5 5 7 5 2 3 7 2 7 5 5 7
 0 9 1 6 5 9 7 4 3 8 0 3 6 4 6 3 2 6 8 8 8 4 6 7 5 2 4 5 3 2 4 6 9 4 5 4 3
 4 6 2 9 0 1 7 2 0 9 6 0 4 2 0 7 9 8 5 4 8 2 8 4 3 7 2 6 9 1 5 1 0 8 2 1 9
 5 6 8 2 7 2 1 5 1 6 4 5 0 9 4 1 1 7 0 8 9 0 5 4 3 8 8 6 5 3 4 4 4 8 8 7 0
 9 6 3 5 2 3 0 8 3 3 1 3 3 0 0 4 6 0 7 7 6 2 0 4 4 2 3 7 8 9 8 6 8 5 6 2 2
 3 1 7 7 8 0 3 3 2 1 5 5 9 1 3 7 0 0 7 0 4 5 9 3 3 4 3 1 8 9 8 3 6 2 1 6 2
 1 7 5 5 1 9 2 8 9 7 2 1 4 9 3 2 6 2 5 9 6 5 8 2 0 7 8 0 5 8 4 1 8 6 4 3 4
 2 0 4 5 8 3 9 1 8 3 4 5 0 8 5 6 3 0 6 9 1 5 2 2 1 9 8 4 3 3 0 7 8 8 1 1 3
 5 5 8 4 9 7 8 4 4 9 0 1 6 9 3 6 1 7 0 6 2 9 9 3 6 1 5 1 8 9 8 4 1 7 2 8 0
 6 2 1 0 6 1 6 5 2 8 6 2 1 4 6 8 2 2 7 5 9 1 9 5 0 2 5 5 6 8 9 5 7 0 5 2 1
 1 2] --预测结果为: [3 8 2 6 6 7 1 9 8 0 2 8 6 6 6 6 4 0 5 8 8 7 8 4 7 5 4 9 2 9 4 7 6 8 9 8 3
 1 0 1 8 6 7 7 1 0 7 0 2 1 9 6 7 9 9 0 9 1 6 3 0 2 3 4 1 9 2 6 9 1 8 6 5 1
 2 8 2 4 9 7 2 3 6 0 9 3 7 5 1 2 0 9 3 1 4 1 4 8 5 4 5 1 7 5 9 0 5 1 4 8 3
 4 8 9 7 9 8 0 4 5 2 5 3 4 8 7 0 6 1 5 3 3 9 5 9 9 5 7 5 6 2 8 6 5 6 1 5 1
 5 9 9 1 3 3 6 1 8 9 2 7 6 7 6 5 6 0 8 8 9 8 8 1 0 4 2 6 3 8 6 7 4 4 6 3 9
 5 3 3 0 7 7 5 7 8 0 7 8 9 6 4 5 0 1 4 6 4 3 3 0 9 5 5 1 3 4 2 1 6 8 9 7 4
 9 3 7 6 2 3 3 1 6 9 3 6 3 7 2 0 7 6 1 1 3 7 3 7 8 5 5 7 5 3 3 7 2 7 5 5 7
 0 9 1 6 5 9 7 4 3 8 0 3 6 4 6 3 1 6 8 8 8 4 6 7 5 2 4 2 3 2 4 6 9 0 5 4 3
 4 6 2 5 0 1 7 2 0 9 6 6 4 2 0 7 9 8 5 7 8 2 8 4 3 7 2 6 7 1 5 9 0 8 2 4 9
 5 6 8 2 7 2 1 5 1 6 4 5 0 9 4 1 2 7 0 5 9 0 5 4 3 8 8 6 5 3 4 4 4 2 8 7 0
 9 6 3 5 2 3 0 8 1 3 1 3 3 0 0 7 6 0 7 7 6 8 0 4 4 8 3 7 8 9 0 6 8 5 6 2 2
 3 1 7 7 3 0 3 3 2 1 5 5 9 1 9 7 0 0 7 0 4 5 8 3 3 4 1 1 8 9 7 9 6 3 1 6 2
 1 7 5 5 4 9 2 8 9 4 2 1 4 1 3 1 6 2 5 9 4 5 1 4 0 7 8 0 5 8 4 1 8 6 2 3 4
 2 0 4 5 8 3 8 1 8 3 4 5 0 8 5 6 3 0 6 9 1 5 1 2 1 9 9 4 8 3 0 7 8 8 1 1 3
 5 5 8 4 9 7 8 4 4 9 0 1 6 9 3 6 1 7 0 4 2 9 5 9 6 1 5 1 1 9 1 4 1 7 2 8 0
 6 2 1 0 6 1 6 5 2 8 6 2 1 4 6 8 2 6 7 5 3 1 9 5 0 2 5 5 6 4 9 5 7 0 8 2 1
 1 2]
CART分类树准确率0.8586

 

python决策树

标签:dig   rac   8 8   load   sklearn   抽取   创建ca   col   and   

原文地址:https://www.cnblogs.com/xiao02fang/p/13513328.html

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