Save the date! Google I/O returns May 18-20 Register now

tff.simulation.models.mnist.create_simple_keras_model

Returns an instance of tf.Keras.Model with just one dense layer.

learning_rate The learning rate to use with the SGD optimizer.

An instance of tf.Keras.Model.