Xtra-Computing / FedTree

A tree-based federated learning system (MLSys 2023)
https://fedtree.readthedocs.io/en/latest/index.html
Apache License 2.0
140 stars 38 forks source link

可以在每一轮后输出测试集的评价指标吗? #71

Open harmony0303 opened 8 months ago

harmony0303 commented 8 months ago

第一个问题是可以在每一轮后输出测试集的评价指标吗? 第二个问题是我设置了bagging=1 ,但是好像输出没有变化,是没有vertical版本的random forest吗?

harmony0303 commented 8 months ago

第二个问题已解决,好像设置bagging=1后,最终输出的会有变化而训练过程中没有。

QinbinLi commented 8 months ago

@harmony0303

目前我们不支持每一轮输出测试集指标,我们会考虑之后加上这个。 我们没有vertical版本的random forest,设置成bagging=1之后,会把所有树的预测结果平均起来作为输出。