Module: tff.learning.programs

Package of methods for compositional federated program logic.


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


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.