使用MLP算法进行k倍交叉验证

问题描述 投票:-1回答:1

我有一个数据集,该数据集分为训练和测试部分。我的任务是训练它并使用k倍交叉验证评估我的模型。我对任务说明有些困惑。据我所知,k-dold交叉验证的重点是通过使用所有数据来训练和测试模型来对有限数据样本进行评估。请告诉我我的算法是否正确:

  1. 连接测试和数据集的训练部分。
  2. 将整个数据集分成k个部分。
  3. 使用MLP在我的数据集的所有部分上训练模型,除了一个,然后在左侧对其进行测试。保留评估分数。
  4. 计算平均评价分数。
machine-learning neural-network k-fold mlp
1个回答
3
投票

是的,您做得对。使用K折交叉验证的全部要点是因为我们的数据有限,并且它确保来自原始数据集的每个观察结果都有机会出现在训练和测试集中。

您提到的步骤:

  1. 将整个数据随机分成k倍(k的值不能太小或太高,理想情况下,根据数据大小,我们选择5到10)。

  2. 然后使用K -1折对模型进行拟合,并使用剩余的K折对模型进行验证。保存分数和错误。

  3. 重复此过程,直到每个K折叠都用作测试集为止。然后取您所记录分数的平均值。这将是模型的性能指标。

对点1进行编辑:较高的K值导致较少的模型偏差,但较大的方差可能会导致过度拟合,其中,较低的K值类似于火车测试拆分方法。因此,我们选择k值在5到10之间。您可以尝试使用这些值以获得更好的性能指标。

© www.soinside.com 2019 - 2024. All rights reserved.