Module: tff.simulation.models.mnist

Stay organized with collections Save and categorize content based on your preferences.

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.