Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tff.simulation.models.mnist.create_simple_keras_model

View source on GitHub

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

tff.simulation.models.mnist.create_simple_keras_model(
    learning_rate=0.1
)

Args:

  • learning_rate: The learning rate to use with the SGD optimizer.

Returns:

An instance of tf.Keras.Model.