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

时间:2017-02-18 16:54:20

标签: python tensorflow mnist

目前我正在寻找一种方法来过滤我的训练数据集(MNIST)中的特定类,以在不同的星座上训练神经网络,例如:仅在第4,5,6类训练网络,然后在0,1,2,3,4,5,6,7,8,9上训练它以评估测试数据集的结果。

我想通过控制台使用参数解析器来选择哪些类应该在我的训练数据集中,这样我就可以将它分成小批量。我想我可以通过标签整理出来,但我现在有点卡住...会很感激任何提示!

问候, 亚历

1 个答案:

答案 0 :(得分:-1)

找到答案我猜...一个热门=真正将标量转换为一个热门矢量:)无论如何感谢您的时间!