Est-il possible d'importer uniquement les images MNIST avec 0 et 1?

Question

Je commence juste avec tensorflow et je veux tester quelque chose seulement sur les 0 et les 1 des images MNIST. Est-il possible d'importer uniquement ces images?

Answer 1

En supposant que vous utilisez

from tensorflow.examples.tutorials.mnist import input_data

Non, il n'y a pas de fonction ou un argument dans ce fichier ... Ce que vous pouvez faire est de charger toutes les données, et de ne sélectionner que les uns et de zéros.