November 2019
Intermediate to advanced
346 pages
9h 36m
English
import tensorflow as tfdef preprocess_observations(data): """Preprocesses MNIST images.""" data = np.array(data, dtype=np.float32) / 255 data = data.reshape(data.shape[0], 28, 28, 1) return datadef preprocess_labels(labels): """Preprocess MNIST labels.""" labels = np.array(labels, dtype=np.int32) labels = tf.keras.utils.to_categorical(labels, num_classes=10)
def load_mnist(): """Loads the MNIST dataset.""" (X_train, y_train), (X_test, y_test) = tf.keras.datasets.mnist.load_data() X_train = preprocess_observations(X_train) X_test = preprocess_observations(X_test) y_train = preprocess_labels(y_train) ...