Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tff.simulation

View source

The public API for experimenters running federated learning simulations.

Modules

datasets module: Datasets for running Federated Learning experiments in simulation.

Classes

class ClientData: Object to hold a dataset and a mapping of clients to examples.

class FilePerUserClientData: A tf.simulation.ClientData that maps a set of files to a dataset.

class FromTensorSlicesClientData: ClientData based on tf.data.Dataset.from_tensor_slices.

class HDF5ClientData: A tff.simulation.ClientData backed by an HDF5 file.

class TransformingClientData: Transforms client data, potentially expanding by adding pseudo-clients.