ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tff.simulation.models.mnist

An example of an MNIST model function for use with TensorFlow Federated.


create_keras_model(...): Returns an instance of tf.keras.Model for use with the MNIST example.

create_simple_keras_model(...): Returns an instance of tf.Keras.Model with just one dense layer.

keras_dataset_from_emnist(...): Converts dataset for use with the output of create_simple_keras_model.