Save the date! Google I/O returns May 18-20 Register now

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.