FLgo学习

https://github.com/WwZzz/easyFL

安装:pip install flgo numba

作者:https://zhuanlan.zhihu.com/p/614218973

quick-start

python 复制代码
import flgo
import flgo.benchmark.mnist_classification as mnist
import flgo.benchmark.partition as fbp
import flgo.algorithm.fedavg as fedavg

# Line 1: Create a typical federated learning task
flgo.gen_task_by_(mnist, fbp.IIDPartitioner(num_clients=5), './my_task')
# Line 2: Running FedAvg on this task
fedavg_runner = flgo.init('./my_task', fedavg, {'gpu': [0,], 'num_rounds':20, 'num_epochs': 1})
# Line 3: Start Training
fedavg_runner.run()

# Show Training Result
import flgo.experiment.analyzer as fea
# Create the analysis plan
analysis_plan = {
    'Selector':{'task': './my_task', 'header':['fedavg',], },
    'Painter':{'Curve':[{'args':{'x':'communication_round', 'y':'val_loss'}}]},
}

fea.show(analysis_plan)

数据异构

  • diversity 类型
  • Dirichlet 分布

讨论两种数据异构类型,对训练准确率的影响

不知道为什么改 num_clients 就会报错,难道只能是100?

有时间再研究下代码

相关推荐
守城小轩1 小时前
从零开始学习Dify-数据库数据可视化(五)
ai·ai客服·ai浏览器
YXWik610 小时前
Linux(centos7)安装 docker + ollama+ deepseek-r1:7b + Open WebUI(内含一键安装脚本)
linux·docker·ai
charlee4415 小时前
行业思考:不是前端不行,是只会前端不行
前端·ai
豌豆花下猫17 小时前
Python 潮流周刊#112:欢迎 AI 时代的编程新人
后端·python·ai
鹿先森AI探索之路17 小时前
本地部署Dify教程
人工智能·docker·ai
连合机器人1 天前
酷暑来袭,科技如何让城市清凉又洁净?
人工智能·ai·有鹿机器人·连合直租·智能清洁专家
努力一点9481 天前
linux系统底层逻辑 开机顺序 ubuntu22.04系统
linux·运维·服务器·ubuntu·ai·gpu算力
智合同(小智)1 天前
娃哈哈遗产继承中的合同智能应用挑战:当家族企业传承遇上法律科技
科技·ai·合同智能应用·智合同·娃哈哈·法律科技·遗产继承
AlfredZhao1 天前
AI再强大,也不如人类员工用的爽?
ai·nl2sql
mpr0xy2 天前
编译支持cuda硬件加速的ffmpeg
ai·ffmpeg·nvidia·cuda