Attend the Women in ML Symposium on December 7 Register now

Module: tff.simulation

Stay organized with collections Save and categorize content based on your preferences.

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 iterative process which constructs datasets on clients.

run_server(...): Runs a gRPC server hosting a simulation component in this process.

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

server_context(...): Context manager yielding gRPC server hosting simulation component.

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'