|View source on GitHub|
Libraries for working with models in federated learning algorithms.
class FunctionalModel: A model that parameterizes forward pass by model weights.
class ModelWeights: A container for the trainable and non-trainable variables of a
class ReconstructionModel: Represents a reconstruction model for use in Tensorflow Federated.
class VariableModel: Represents a variable-based model for use in TensorFlow Federated.
load_functional_model(...): Deserializes a TensorFlow SavedModel at
path to a functional model.
model as a TensorFlow SavedModel to
save_functional_model(...): Serializes a
FunctionalModel as a