假设有训练数据 train_images和标签train_labels,已经转换为np.ndarray类型,有两种方式实现训练图像数据和标签的随机打乱。
- 方式一,固定随机种子
# 在每轮迭代开始前,将训练数据随机打乱
seed_id = np.random.randint(1,100)
np.random.seed(seed_id)
np.random.shuffle(train_images)
np.random.seed(seed_id)
np.random.shuffle(train_labels)
- 方式二,先随机打乱索引号,再索引数据和标签
# 在每轮迭代开始前,将训练数据随机打乱
index = [i for i in range(train_images.shape[0])]
np.random.shuffle(index) # 打乱索引
image = train_images[index]
label = train_label[index]
网友评论