-
Notifications
You must be signed in to change notification settings - Fork 914
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Xgboost-quickstart example #2612
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Example looks great. To further simplify, I'd suggest to:
- Remove
strategy.py
and use the strategy inflwr.server.strategy
- Remove
dataset.py
and move the code intoclient.py
Co-authored-by: Daniel J. Beutel <[email protected]>
Co-authored-by: Daniel J. Beutel <[email protected]>
# Let's use the first partition as an example | ||
partition = fds.load_partition(idx=0, split="train") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we add the possibility to load a specific partition by providing --node-id 0
or --node-id 1
? And then use IidPartitioner(num_partitions=2)
above?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ok, i'll pull back the arguments parser only for --node-id
Issue
Description
We need a simpler quick-start example for XGBoost, which allows users to get a quick idea of how to use XGBoost in Flower.
Proposal
In this example, we simply the data partitioning part, and remove the customised experimental settings. We only use two clients for the training. The evaluation is conducted on clients' hold-out test set.
Note
After bagging strategy has been moved to the core framework, this PR, we need to change the server code here to use the strategy from the core framework.