Missed TensorFlow World? Check out the recap. Learn more

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.