如何实现我自己的联合算法Usinsg tensorflow联合

时间:2019-12-20 18:53:53

标签: tensorflow federated

我已经阅读了tensorflow.org上有关tensorflow联合的所有文档,但是我不确定如何实现自己的联合算法。例如,我有一个编译的keras模型,我知道如何将其转换为tff.computation。似乎为了构建联合算法,应该构建一个iterative_process。有人可以在这方面帮助我吗?

非常感谢您

2 个答案:

答案 0 :(得分:1)

您可以在tensorflow联合存储中读取文件 optimizer_utils.py ,该文件位于 python / learning / framework /

更多详细信息,请参见link

名为one_round_computation的函数在一轮中实现联合算法的编排逻辑。最重要的是使用tff.federated_map,tff.federated_broadcast,tff.federated_collect ...来构建抽象语法树。

答案 1 :(得分:0)

您应该使用以下代码:

tff.learning.from_compiled_keras_model(kera_model, dummy_batch)

相关问题