Running example code from Tensorflow documentation, pertaining to Tensorflow-Federated, results in AttributeError

Question

I am trying to run the code given by Tensorflow in their official documentation, pertaining to Tensorflow-Federated. The code is as follows:

import tensorflow as tf
import tensorflow_federated as tff

training_process = tff.learning.algorithms.build_weighted_fed_avg(get_dense_nn, client_optimizer_fn=lambda: tf.keras.Sequential.SGD(learning_rate=client_lr),                                                                 server_optimizer_fn=lambda:tf.keras.optimizers.SGD(learning_rate=server_lr))

However, I am getting the following error:

AttributeError: module 'tensorflow_federated.python.learning' has no attribute 'from_keras_model'

Am I doing something wrong or is the documentation wrong?

Is your python version 3.9 or later? As per the documentation page Docs — fam-woodpecker
– fam-woodpecker, Commented Mar 18, 2024 at 1:27

mkrieger1 · Accepted Answer · 2024-03-18 15:05:40Z

0

The error message indicates that tensorflow_federated.python.learning.from_keras_model is attempting to be used, however this API does not exist.

In newer versions of TensorFlow Federated, the API has moved to tff.learning.models.from_keras_model.

edited Mar 18, 2024 at 15:05

mkrieger1

24.2k7 gold badges68 silver badges84 bronze badges

answered Mar 18, 2024 at 12:33

Zachary Garrett

2,93115 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Running example code from Tensorflow documentation, pertaining to Tensorflow-Federated, results in AttributeError

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related