ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tff.learning.models

Libraries for working with models in Federated Learning algorithms.

Classes

class FunctionalModel: A model that parameterizes forward pass by model weights.

Functions

load(...): Deserializes a TensorFlow SavedModel at path to a tff.learning.Model.

load_functional_model(...): Deserializes a TensorFlow SavedModel at path to a functional model.

model_from_functional(...): Converts a FunctionalModel to a tff.learning.Model.

save(...): Serializes model as a TensorFlow SavedModel to path.

save_functional_model(...): Serializes a FunctionalModel as a tf.SavedModel to path.