多维数据和标签一起打乱 shuffle

原文链接: 多维数据和标签一起打乱 shuffle

上一篇: kaggle 手写数字识别 卷积网咯

下一篇: 全连接网络 椭圆区域划分

使用sklearn 自带的 shuffle 函数

首先创建多维数组，使用图像基本数据格式 N*W*H*C

使用shuffle函数同时打乱标签和数据，并保持标签与数据的对应关系

from sklearn.utils import shuffle
import numpy as np

data = np.stack([
    np.full((2, 2, 1), i)
    for i in range(10)
])
label = np.arange(10)

print(data.shape, label.shape)
print(data[:3], label[:3])

n_data, n_label = shuffle(data, label)
print(n_data[:3], n_label[:3])

(10, 2, 2, 1) (10,)
[[[[0]
   [0]]

  [[0]
   [0]]]


 [[[1]
   [1]]

  [[1]
   [1]]]


 [[[2]
   [2]]

  [[2]
   [2]]]] [0 1 2]
[[[[0]
   [0]]

  [[0]
   [0]]]


 [[[5]
   [5]]

  [[5]
   [5]]]


 [[[7]
   [7]]

  [[7]
   [7]]]] [0 5 7]