从MNIST数据集中选择每个类的特定数量可以通过以下步骤实现:
from tensorflow.keras.datasets import mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
num_classes = 10
class_samples = [100, 200, 150, 120, 180, 90, 80, 60, 100, 150] # 每个类别需要选择的样本数量
class_data = [[] for _ in range(num_classes)] # 创建一个空列表用于存储每个类别的样本
for i in range(len(x_train)):
label = y_train[i]
if len(class_data[label]) < class_samples[label]:
class_data[label].append(x_train[i])
在上述代码中,class_samples
列表指定了每个类别需要选择的样本数量。class_data
列表用于存储每个类别的样本。
完成上述步骤后,你将得到一个包含每个类别特定数量样本的数据集。这个数据集可以用于训练、验证或测试机器学习模型。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云