kaggle挑战赛——不同肺炎疾病三分类

2022-08-20 11:36:39 浏览数 (1)

今天将分享kaggle系列挑战赛之病毒性肺炎,细菌性肺炎多分类的完整实现过程,为了方便大家学习理解整个流程,将整个流程步骤进行了整理,并给出详细的步骤结果。感兴趣的朋友赶紧动手试一试吧。

一、比赛介绍

数据下载来自kaggle挑战赛,数据组成由正常肺,病毒性肺炎,细菌性肺炎组成。可以下载训练数据和测试数据,测试数据一共300例,训练数据一共有4148例。

二、技术路线

1、加载csv文件,读取X光图像和对应标签值。

2、将训练数据分成9:1分成网络训练数据,验证数据,测试集。

3、对所有的图像进行统一缩放处理,大小统一大小是256x256,图像归一化,用均值是0和方差是1的方式进行归一化。

4、网络采用的是VGG16网络结构,损失是交叉熵,学习率是0.001,droupout是0.5,epoch是20,batchsize是64。

5、训练结果

训练损失函数结果:

训练精度函数结果:

验证损失函数结果:

验证精度函数结果:

6、测试结果

三组ROC曲线及数值

标签0与其他标签

标签1与其他标签

标签2与其他标签

混淆矩阵:

分类指标:

precision recall f1-score support

0.0 1.00 0.60 0.75 100

1.0 0.96 0.78 0.86 100

2.0 0.62 0.99 0.76 100

avg / total 0.86 0.79 0.79 300

为了方便大家更高效地学习,后期会将相关项目代码更新到github上,点击原文链接即可访问官方比赛网址,可以直接下载训练和测试数据。

如果大家觉得这个项目还不错,希望大家给个Star并Fork,可以让更多的人学习。如果有任何问题,随时给我留言我会及时回复的。

0 人点赞