Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

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.