tff.simulation.models.mnist.create_simple_keras_model

Stay organized with collections Save and categorize content based on your preferences.

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.