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.