Pre-trained Model and training_mode changes - Githubissues

NVIDIA / NVFlare

NVIDIA Federated Learning Application Runtime Environment

https://nvidia.github.io/NVFlare/

Apache License 2.0

630 stars 179 forks source link

Pre-trained Model and training_mode changes #2793

Closed nvidianz closed 2 months ago

nvidianz commented 2 months ago

Description

Added support for pre-trained model in XGBoost. The model must reside in custom/model.json.
Changed training_mode to split_mode and secure_training to align with XGBoost parameters.
Merged changes made to main and changed all the examples to use split_mode.

Types of changes

[x] Non-breaking change (fix or new feature that would not break existing functionality).
[ ] Breaking change (fix or new feature that would cause existing functionality to change).
[ ] New tests added to cover the changes.
[ ] Quick tests passed locally by running ./runtest.sh.
[ ] In-line docstrings updated.
[ ] Documentation updated.

nvidianz commented 2 months ago

have you test this ? to have a pre-trained model in the json format ? custom/model.json

Yes. With the model there, the AUC improved a little bit.