| from tensorflow.keras.utils import to_categorical | |
| # Normalize the images | |
| train_images = train_images / 255.0 | |
| test_images = test_images / 255.0 | |
| # Reshape them for our CNN model | |
| train_images = train_images.reshape(60000,28,28,1) | |
| test_images = test_images.reshape(10000,28,28,1) | |
| print(train_images.shape) | |
| # Encode labels to one-hot | |
| train_labels = to_categorical(train_labels) | |
| test_labels = to_categorical(test_labels) | |
| print(train_labels[0]) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment