tensorflow.keras.Model compile fit evaluate应用

tensorflow.keras.Model compile fit evaluate应用

import  tensorflow as tf
from    tensorflow.keras import datasets, layers, optimizers, Sequential, metrics

数据预处理

def preprocess(x, y):
    """
    x is a simple image, not a batch
    """
    x = tf.cast(x, dtype=tf.float32) / 255.
    x = tf.reshape(x, [28*28])
    y = tf.cast(y, dtype=tf.int32)
    y = tf.one_hot(y, depth=10)
    return x,y

batchsz = 128
(x, y), (x_val, y_val) = datasets.mnist.load_