def next_batch(train_data, train_target, batch_size):
index = [ i for i in range(0,len(train_target)) ]
np.random.shuffle(index);
batch_data = [];
batch_target = [];
for i in range(0,batch_size):
batch_data.append(train_data[index[i]]);
batch_target.append(train_target[index[i]])
return batch_data, batch_target
批量读取数据next_batch()简单实现
最新推荐文章于 2021-07-09 23:43:27 发布