tff.learning 模块

tff.learning 模块

用于使用联合学习算法的模型开发的公共API

模块

framework模块：面向开发联合学习算法的贡献者的公共API

类

BatchOutput类： 保存tf .learning. model输出的结构。

Model类： 表示TensorFlow联邦中使用的模型。

TrainableModel类： 带有用于(本地)培训的附加方法的模型。

函数

assign_weights_to_keras_model(…):将TFF权重的嵌套结构分配给Keras模型。

build_federated_averaging_process(…):使用联邦平均构建TFF计算用于优化。

build_federated_evaluation(…):为给定模型的联邦评估构建TFF计算。

build_federated_sgd_process(…): 使用联邦SGD构建TFF计算用于优化。

from_compiled_keras_model(…):为一个示例迷你批处理构建一个 tf.learning.Model 建模。

from_keras_model(…):为一个示例迷你批处理构建一个tf.learning.Model 建模。

state_with_new_model_weights(…):返回一个带有更新的模型权重的服务器状态。