如何在 MATLAB 中重新排序神经网络的测试数据集？

Question

我在一个文件夹中有 300 张图像，用作验证神经网络的数据集。

每张图片都有一个1 或0 的标签。我将图像转换为 MATLAB 中的矩阵，并将标签转换为 1 或 0 的 300x1 向量，该向量对应于矩阵中的每一行图像。

我注意到标签不平衡，这会影响网络。如何重新排序标签，同时为每行图像保持相同的零或一指示？

这是我尝试过的：

我将 1 和 0 分别拆分为一个向量，现在我希望能够在 12 行中连续放置至少两个 1 和 10 个零，但我不知道如何操作。我用它来分割一和零：

ones = labels(labels(:,1)==1,:)
zeros = labels(labels(:,1)==0,:)

如何重新排列标签以消除标签不平衡的问题？

Answer 1

这将创建一个索引向量，该向量应均匀分布您的两组标签，以便它们在每个连续的 12 个标签组中大致平衡（请注意，这假设您的数据均匀划分为 12 个组，它在您的示例）：

index = [find(labels(:, 1) == 1); find(labels(:, 1) == 0)];
index = reshape(reshape(index, [25 12]).', [], 1);

然后，您将使用此索引对标签数组的行和任何相关数据（如图像数据数组）重新排序：

labels = labels(index, :);
otherData = otherData(index, :);