Missed TensorFlow World? Check out the recap. Learn more

Module: tff.simulation.datasets

View source on GitHub

Datasets for running Federated Learning experiments in simulation.

Modules

emnist module: Libraries for the federated EMNIST dataset for simulation.

shakespeare module: Libraries for the Shakespeare dataset for federated learning simulation.

stackoverflow module: Libraries for the Stackoverflow dataset for federated learning simulation.

Functions

build_dataset_mixture(...): Build a new dataset that probabilistically returns examples.

build_single_label_dataset(...): Build a new dataset that only yields examples with a particular label.

build_synthethic_iid_datasets(...): Constructs an iterable of IID clients from a tf.data.Dataset.