文章目录
- 项目说明
- 代码实现
-
- 1、导入需要的库
- 2、定义相关的类
-
- 2.1 Data
- 2.2 Aggregator
- 2.3 Model
- 2.4 Bank
- 3、导入并处理原始数据
- 4、联邦学习
-
- 4.1 划分数据并重采样
- 4.2 模型训练
-
- 4.2.1 仅使用 Data_Global 训练服务器模型
- 4.2.2 联邦学习训练服务器模型
- 4.3 模型测试
- 5、非联邦学习模型对比
-
- 5.1 建立模型
- 5.2 模型训练
-
- 5.2.1 使用不平衡数据训练
- 5.2.2 使用平衡数据训练