search

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

Log prediction score for vertical FL #39

Closed wxli0 closed 2 years ago

wxli0 commented 2 years ago

I wanted to see the prediction score and predicted labels of the test dataset in vertical FL. After adding some LOG(INFO) to the main.py, GBDT::predict_score_vertical and GBDT::predict_raw, I found the the logs after model.host_data() cannot be printed out to console.

Could someone please suggest how to resolve the issue? Thank you very much!

QinbinLi commented 2 years ago

Hi @wxli0 ,

predict_score_vertical calls predict_raw_vertical instead of predict_raw. Maybe you put LOG(INFO) in a wrong place?

wxli0 commented 2 years ago

Thank you very much for your help! I made some changes in your code which might break the flow. I reverted back to the original copy, and the problem has been resolved. Thanks again!