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?

有时间再研究下代码

相关推荐
xuhe23 小时前
关于nano banana模型实践的思考
ai
Honeysea_704 小时前
容器的定义及工作原理
人工智能·深度学习·机器学习·docker·ai·持续部署
zandy10116 小时前
LLM与数据工程的融合:衡石Data Agent的语义层与Agent框架设计
大数据·人工智能·算法·ai·智能体
seegaler6 小时前
WrenAI:开源革命,重塑商业智能未来
人工智能·microsoft·ai
爱吃烤鸡翅的酸菜鱼6 小时前
飞算JavaAI炫技赛:电商系统开发全流程实战解析
ai·java开发·飞算javaai炫技赛
码界奇点8 小时前
豆包新模型矩阵与PromptPilot构建企业级AI开发的体系化解决方案
人工智能·线性代数·ai·语言模型·矩阵·硬件工程
SheldonChang9 小时前
Onlyoffice集成与AI交互操作指引(Iframe版)
java·人工智能·ai·vue·onlyoffice·postmessage
FutureUniant9 小时前
GitHub每日最火火火项目(9.5)
人工智能·microsoft·计算机视觉·ai·github
CoderJia程序员甲9 小时前
GitHub 热榜项目 - 日榜(2025-09-05)
ai·开源·github·ai编程·github热榜
z千鑫10 小时前
【模型比对】Gemini 2.5 Pro 与 Claude Sonnet 4 结构化数据对比报告 + API KEY的使用教程
人工智能·gpt·ai·语言模型·aigc