PyTorch-通过ImageFolder加载数据时如何使用k倍交叉验证？

Question

我的图像数据存储在文件系统中，并通过ImageFolder的{​​{1}}数据加载器馈送到我的卷积神经网络中。因此，PyTorch，training和validation数据被手动拆分到文件系统上的不同文件夹中。那么，在使用test时如何应用k-fold cross validation？

Answer 1

您可以使用data.ConcatDataset将当前具有的固定火车/停车/测试折叠合并到单个数据集中。然后，您可以使用data.Subset一遍又一遍地将单个数据集随机分成不同的折叠。