【发布时间】:2018-08-10 05:58:20
【问题描述】:
我使用 CNN 构建了一个 3D 图像分类模型用于我的研究。我只有 5000 张图像,使用 4500 张图像进行训练,500 张图像用于测试集。 我为训练尝试了不同的架构和参数 F1 分数和训练集的准确率高达 0.9。幸运的是,我不必花费大量时间来找到这些设置以获得高精度。
现在我将这个模型应用于测试集,我得到了一个非常令人满意的预测,F1 分数为 0.8~0.85。
我的问题是,是否有必要进行验证?那时我参加机器学习课程时,我被教导使用验证集来调整超参数。我没有进行 k 折交叉验证的一个原因是因为我没有太多数据并且想使用尽可能多的训练数据。我的模型在测试集上显示出相当好的预测。只要准确率/f1 分数/ROC 足够好,我的模型还能让人信服吗?或者我可以尝试仅通过进行 k 折交叉验证而不单独对测试集进行制作和测试来说服人们吗?
谢谢!
【问题讨论】:
-
鉴于您的测试集非常小(500 张图像),我会进行交叉验证。您没有太多数据的事实凸显了交叉验证的必要性。相反,如果您有足够大的测试集,则可以跳过 CV 过程。
标签: validation machine-learning cross-validation