Flwr框架下使用sklearn梯度提升树的参数初始化问题
Flwr框架下联邦梯度提升树实现的参数适配问题
- 近期基于Flwr框架落地联邦梯度提升树方案,首先验证了Flwr官方提供的
sklearn-logreg-mnist参考示例可正常运行:该示例采用sklearn.linear_model.LogisticRegression模型完成MNIST分类任务,在本地运行环境无兼容问题。 - 直接将示例中的逻辑回归模型替换为
sklearn.ensemble.GradientBoostingClassifier时触发运行报错。
核心待解决问题:梯度提升分类器需要初始化哪些参数,才能等效匹配逻辑回归中
LogisticRegression.coef_承担的模型参数交互功能?
- 待调试出可稳定运行的版本后,会将完整实现提交到Flwr官方示例库,面向所有开发者开放参考。
内容的提问来源于stack exchange,提问作者Elias
相关产品推荐
相关产品推荐

