onlyzdd / ecg-diagnosis

Deep learning for 12-lead ECG interpretation
123 stars 34 forks source link

有关于十折交叉验证的问题 #7

Open Linnnhaiii opened 1 year ago

Linnnhaiii commented 1 year ago

作者您好,我在您发表的论文《Interpretable deep learning for automatic diagnosis of 12-lead electrocardiogram》补充材料中的Training and Evaluation部分看到您提到应用了十折交叉验证的方法(一共进行十轮实验,每轮都将8fold用于训练,1fold用来验证,1fold用于测试,最后取10轮的平均值作为结果),但是我在代码中没有看到相应的实现,不知道是不是我理解错了(我所看到的是只进行了一轮划分),希望能得到您的回复,万分感谢。

rdyan0053 commented 1 year ago

好像确实是这样,运行一次train.py,产生一个模型,比如resnet34_CPSC_all_42.pth,42就是随机数种子,确保数据集的划分可以重现。

所以如果需要10折交叉验证,是不是要运行10次train.py 应该每次运行的时候,先修改一下训练集,验证集,测试集,然后再运行