联邦学习(二):卷积神经网络通过底层 API 实现 FedAvg 联邦算法

文章目录

  • 1、导入需要的库
  • 2、导入并处理数据
  • 3、定义批次数据类型
  • 4、定义模型类型
    • 4.1 得到每一层神经网络的参数形状
    • 4.2 得到模型类型
    • 4.3 建立模型
  • 4、定义计算一个批次损失的函数
  • 5、定义通过一个批次的损失更新网络参数的函数
  • 6、定义通过一个客户端的损失更新网络参数的函数
  • 7、定义通过所有客户端的损失更新网络参数的函数

1、导入需要的库

import collections

import numpy as np
import tensorflow as tf
import tensorflow_federated as tff

你可能感兴趣的:(联邦学习,深度学习,python,python,深度学习,联邦学习)