Skip to content
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

Merged
merged 10 commits into from Nov 17, 2023
Merged

Xgboost-quickstart example #2612

merged 10 commits into from Nov 17, 2023

Conversation

yan-gao-GY
Copy link
Contributor

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.

@jafermarq jafermarq added the Examples Add or update a Flower example label Nov 16, 2023
Copy link
Member

@danieljanes danieljanes left a 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 in flwr.server.strategy
  • Remove dataset.py and move the code into client.py

examples/xgboost-quickstart/README.md Show resolved Hide resolved
examples/xgboost-quickstart/README.md Outdated Show resolved Hide resolved
Comment on lines 52 to 53
# Let's use the first partition as an example
partition = fds.load_partition(idx=0, split="train")
Copy link
Member

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?

Copy link
Contributor Author

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

@danieljanes danieljanes enabled auto-merge (squash) November 17, 2023 12:43
@danieljanes danieljanes merged commit 5cf5c82 into main Nov 17, 2023
27 checks passed
@danieljanes danieljanes deleted the xgboost-quickstart branch November 17, 2023 12:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Examples Add or update a Flower example
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants