Comment sous-créer un ensemble de données MNIST?

Question

Je suis un tutoriel sur Generative Adversarial Network pour TensorFlow. Le tutoriel utilise un jeu de données MNIST pour former le modèle. Je veux réduire la taille de l'entrée pour que mon programme s'exécute plus vite mais je n'ai aucune idée de comment obtenir un sous-ensemble de l'ensemble de données MNIST que j'utilise. Ci-dessous, le code I utilisé pour extraire l'ensemble de données:

from tensorflow.examples.tutorials.mnist import input_data 
mnist = input_data.read_data_sets("MNIST_data/") 

Answer 1

Il y a une méthode

mnist.next_batch(batchsize) 

qui extrait un échantillon aléatoire de batchsize longueur de la rame.

Si vous ne voulez pas quelque chose au hasard, vous pouvez y accéder par

x = mnist.train.images[start_batch:end_batch] 
y = mnist.train.labels[start_batch:end_batch] 

ou similaire avec mnist.test pour l'ensemble de test.