Module: tff.learning.models

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

Libraries for working with models in federated learning algorithms.


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


functional_model_from_keras(...): Converts a tf.keras.Model to a tff.learning.models.FunctionalModel.

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.