|View source on GitHub|
Libraries for working with models in federated learning algorithms.
class BatchOutput: A structure that holds the output of a
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.
functional_model_from_keras(...): Converts a
tf.keras.Model to a
load(...): Deserializes a TensorFlow SavedModel at
path to a
load_functional_model(...): Deserializes a TensorFlow SavedModel at
path to a functional model.
model_from_functional(...): Converts a
FunctionalModel to a
model as a TensorFlow SavedModel to
save_functional_model(...): Serializes a
FunctionalModel as a
weights_type_from_model(...): Creates a
tff.Type from a
tff.learning.models.VariableModel or callable that constructs a model.