Module: tff.simulation

Libraries for running TensorFlow Federated simulations.

Modules

baselines module: Libraries for constructing baseline learning tasks suitable for simulation.

datasets module: Datasets for running TensorFlow Federated simulations.

models module: Models for running TensorFlow Federated simulations.

Functions

build_uniform_sampling_fn(...): Builds the function for sampling from the input iterator at each round.

compose_dataset_computation_with_computation(...): Builds a new tff.Computation which constructs datasets on clients.

compose_dataset_computation_with_iterative_process(...): Builds a new iterative process which constructs datasets on clients.

compose_dataset_computation_with_learning_process(...): Builds a new learning process which constructs datasets on clients.

run_training_process(...): Runs a federated training_process.

EVALUATION_METRICS_PREFIX 'evaluation/'
EVALUATION_TIME_KEY 'evaluation_time_in_seconds'
ROUND_NUMBER_KEY 'round_number'
ROUND_TIME_KEY 'round_time_in_seconds'
TRAINING_TIME_KEY 'training_time_in_seconds'