AttributeError:模块“tensorflow_federated.python.learning”没有属性“算法”

Question

我正在尝试运行由Tensorflow 在官方文件中提供的代码，该代码属于Tensorflow联邦。守则如下：

import tensorflow as tf
import tensorflow_federated as tff

def model_fn():
  model = tf.keras.models.Sequential([
      tf.keras.layers.Dense(10, tf.nn.softmax, input_shape=(784,),
                            kernel_initializer='zeros')
  ])

trainer = tff.learning.algorithms.build_weighted_fed_avg(
  model_fn,
  client_optimizer_fn=lambda: tf.keras.optimizers.SGD(0.1))

但是，我得到了以下错误：

Traceback (most recent call last):
  File "G:/pythonproject2/main.py", line 43, in <module>
    trainer = tff.learning.algorithms.build_weighted_fed_avg(
AttributeError: module 'tensorflow_federated.python.learning' has no attribute 'algorithms'

有人能帮帮我吗？

Answer 1

我发现了这个问题，tensorflow_federated需要Python 3.9+。由于我使用的是Python3.7，所以遇到了上述错误。

来自Tensorflow联邦文档：

​