sklearn实现决策树对乳腺癌数据集的分类并使用五折交叉验证(上)

关于Breast Cancer Wisconsin (Diagnostic) Data Set 乳腺癌数据集的基本情况可以看我之前发的博客:https://blog.csdn.net/weixin_42305378/article/details/106058697

这边主要是使用sklearn的决策树算法对乳腺癌数据集进行分类,并用五折交叉验证评估

以前我们是直接将数据分割成70%的训练数据和测试数据,现在我们利用5折交叉验证分割数据,首先将数据分为5组,然后再从5组数据之中选择不同数据进行训练。

from sklearn import tree
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.model_selection import cross_val_score
#加载数据集
cancer =datasets.load_breast_cancer()
# pt=datasets.load_sample_images()  输出图片
# print(pt)
cancer_X=cancer.data     #数据
print(cancer_X.shape)
cancer_Y=cancer.target   #标签
print(cancer_Y.shape)
#使用决策分类树
clf = tree.DecisionTreeClassifier(criterion="entropy" )
#引入五折交叉验证
scores=cross_val_score(clf,cancer_X, cancer_Y,cv=5,scoring='accuracy')#评分方式为accuracy
print(scores)       #会得到五组的评分结果
print(scores.mean())#平均评分结果

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值