Запуск примера кода из документации Tensorflow, относящегося к Tensorflow-Federated, приводит к AttributeError. ⇐ Python

[Anonymous]

Я пытаюсь запустить код, указанный 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'

Я делаю что-то не так или документация неверна?

Подробнее здесь: https://stackoverflow.com/questions/781 ... orflow-fed