GitHub

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

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2023-03-17 UTC.