Module: tff.simulation.datasets

See Stable See Nightly

Datasets for running Federated Learning simulations.

Modules

cifar100 module: Libraries for the federated CIFAR-100 dataset for simulation.

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

gldv2 module: Libraries for the federated Google Landmark v2 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.