tff.simulation.models.mnist.create_keras_model

View source on GitHub

Returns an instance of tf.keras.Model for use with the MNIST example.

This code is based on the following target, which unfortunately cannot be imported as it is a Python binary, not a library:

https://github.com/tensorflow/models/blob/master/official/r1/mnist/mnist.py

compile_model If True, compile the model with a basic optimizer and loss.

A tf.keras.Model.