tff.learning.from_keras_model

View aliases

Main aliases

`tff.learning.models.from_keras_model`

tff.learning.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.Model

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