tff.simulation.models.mnist.create_simple_keras_model

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`.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-09-20 UTC.

Args