Module: tff.learning.programs

Package of methods for compositional federated program logic.

Classes

class EvaluationManager: A manager for facilitating multiple in-progress evaluations.

class ProgramState: A structure representing the program state.

Functions

extract_and_rewrap_metrics(...): Extracts a sub-structure and re-wraps it with a new prefix.

train_model(...): Runs specified rounds of training and optionally evaluates the model.

train_model_with_vizier(...): Trains and tunes a federated model using Vizier.