Skip to content

Add general XGBoost demo notebook#7601

Merged
rapids-bot[bot] merged 12 commits intorapidsai:mainfrom
dantegd:xgboost-notebook-demo
Jan 6, 2026
Merged

Add general XGBoost demo notebook#7601
rapids-bot[bot] merged 12 commits intorapidsai:mainfrom
dantegd:xgboost-notebook-demo

Conversation

@dantegd
Copy link
Copy Markdown
Member

@dantegd dantegd commented Dec 11, 2025

Adds an initial version of a notebook demonstrating GPU-accelerated XGBoost training with RAPIDS. The notebook walks through simulating/loading tabular data, configuring XGBoost for GPU execution using device='cuda', and training a binary classification model.

Notes

  • Uses XGBoost 3.1 compatible parameters
  • Requires CUDA-capable GPU

@review-notebook-app
Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@github-actions github-actions Bot added the Cython / Python Cython or Python issue label Dec 11, 2025
@dantegd dantegd added doc Documentation non-breaking Non-breaking change labels Dec 11, 2025
Copy link
Copy Markdown
Contributor

@csadorf csadorf left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm in favor of running this within our CI, but I think it would be confusing to users that we have an XGBoost demo notebook within our notebooks/ directory that is generally documented as providing examples for cuML. Can we place the notebook somewhere else or maybe put it into a (hidden) sub-directory that provides a brief explanation as to why we include this with the repository?

I think placing it somewhere within the ci/ directory might be more appropriate. We can easily adjust the ci/test_notebooks.sh directory to deal with that, e.g., by explicitly listing it or just moving/linking it into the notebooks/ directory right before we execute the tests.

@dantegd dantegd requested a review from a team as a code owner December 15, 2025 21:30
@dantegd dantegd requested a review from AyodeAwe December 15, 2025 21:30
@github-actions github-actions Bot added ci and removed Cython / Python Cython or Python issue labels Dec 15, 2025
@trivialfis
Copy link
Copy Markdown
Member

Please consider using the quantile DMatrix instead

@dantegd
Copy link
Copy Markdown
Member Author

dantegd commented Jan 6, 2026

/merge

@rapids-bot rapids-bot Bot merged commit de15182 into rapidsai:main Jan 6, 2026
107 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci doc Documentation non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants