选择特定的MNIST类来训练神经网络

Question

目前，我正在寻找一种方法，从我的训练数据集（MNIST）中筛选出特定的类来训练不同星座的神经网络，例如，只在4、5、6类训练一个网络，然后在0,1,2,3,4,5,6,7,8,9上训练它，以评估测试数据集的结果。在

我想用一个参数解析器通过控制台来选择哪些类应该在我的训练数据集中，这样我就可以把它分成小批量。我想我可以通过标签进行分类，但我现在有点卡住了。。。如有任何提示，将不胜感激！！！在

您好， 亚历克斯

Answer 1

我猜找到了答案。。。one hot=True将标量转换为one-hot向量：）无论如何，谢谢您的时间！在