tff.learning.models.from_keras_model

tff.learning.models.from_keras_model(
    keras_model: tf.keras.Model,
    loss: Loss,
    input_spec,
    loss_weights: Optional[list[float]] = None,
    metrics: Optional[Union[list[tf.keras.metrics.Metric], list[Callable[[], tf.keras.
        metrics.Metric]]]] = None
) -> tff.learning.models.VariableModel

Used in the notebooks

Used in the tutorials
Building Your Own Federated Learning Algorithm Composing Learning Algorithms Use TFF optimizers in custom iterative process Federated Learning for Image Classification Federated Learning for Text Generation