运行 Tensorflow 文档中与 Tensorflow-Federated 有关的示例代码，会导致 AttributeError

Question

我正在尝试运行 Tensorflow 在其官方文档中给出的与 Tensorflow-Federated 有关的代码。代码如下：

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))

但是，我收到以下错误：

AttributeError：模块“tensorflow_federated.python.learning”没有属性“from_keras_model”

是我做错了什么还是文档有误？

Answer 1

代码片段似乎不包含导致错误的代码，但错误消息显示

tensorflow_federated.python.learning.from_keras_model

正在尝试使用，但此 API 不存在。

在 TensorFlow Federated 的新版本中，API 已移至

tff.learning.models.from_keras_model

。