/mnist-federated-learning

这是一个基于Minist数据集的横向联邦学习实现

Primary LanguagePythonMIT LicenseMIT

简介

这是一个基于Minist数据集的横向联邦学习实现

其主要代码来源于stijani/tutorial

数据下载:kaggle.com/scolianni/mnistasjpg

运行

  • 下载数据集并解压
  • 修改mnist federated learning.py line29,指定数据集
  • 安装环境
pip install -r requirements.txt
  • 运行
python mnist federated learning.py