Watch talks from the 2019 TensorFlow Dev Summit Watch now

Module: tff.learning

Defined in learning/__init__.py.

The public API for model developers using federated learning algorithms.

Modules

framework module: The public API for contributors who develop federated learning algorithms.

Classes

class BatchOutput: A structure that holds the output of a tff.learning.Model.

class Model: Represents a model for use in TensorFlow Federated.

class TrainableModel: A Model with an additional method for (local) training.

Functions

build_federated_averaging_process(...): Builds the TFF computations for optimization using federated averaging.

build_federated_evaluation(...): Builds the TFF computation for federated evaluation of the given model.

build_federated_sgd_process(...): Builds the TFF computations for optimization using federated SGD.

from_compiled_keras_model(...): Builds a tff.learning.Model for an example mini batch.

from_keras_model(...): Builds a tff.learning.Model for an example mini batch.

keras_weights_from_tff_weights(...): Converts TFF's nested weights structure to flat weights.

state_with_new_model_weights(...): Returns a ServerState with updated model weights.