N_IMAGES = 25 images, labels = zip(*[(image, label) for image, label in [train_data[i] for i in range(N_IMAGES)]]) classes = test_data.