Missed TensorFlow World? Check out the recap. Learn more

Module: tff.simulation.models.mnist

View source on GitHub

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

Functions

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.