sklearn实现决策树对乳腺癌的分类诊断(下)

本文介绍了一个使用决策树对乳腺癌四分类问题进行诊断的案例。训练和测试数据集已预先划分,包含超声波图像和临床诊断数据。通过训练决策树模型,预测病人乳腺癌的类型,并将预测结果以CSV文件形式输出。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

前面我们做的关于决策树的代码实现是对一个数据集进行划分为训练集和测试集,这篇使用的训练集和测试集是已经帮你划分好的存放在二个文件里的。

这里我们使用的训练集和测试集都是关于乳腺癌的数据集,但是和前面我们使用的乳腺癌数据集有点不一样,前面的是二分类的问题:良性和恶性,我们这是四分类问题,类别是乳腺癌的四个类型(训练集train/feats.csv的最后一列)。

下面是我们的测试集和训练集:其中images文件里包含了好多个子文件夹,每个子文件夹里又分别有好多张病人的乳腺癌超声波诊断图像,feats.csv里存放的是病人的乳腺癌临床诊断数据。

训练集train/feats.csv中的部分临床诊断数据:

测试集test/feat.csv 中的部分临床诊断数据:

我们现在使用决策树进行分类预测,这里的id是病人的id,预测病人得的是那种类型的乳腺癌,最后输出病人id和对应的乳腺癌种类的csv文件。

#决策树分类 https://blog.youkuaiyun.com/bjjoy2009/articl
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值