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

Update benchmarking notebook #262

Merged
merged 5 commits into from
Aug 7, 2023
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 16 additions & 2 deletions source/examples/xgboost-rf-gpu-cpu-benchmark/notebook.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,29 @@
{
"cell_type": "markdown",
"id": "a51c95d1-b447-4f1b-9571-cf597ca93ef4",
"metadata": {},
"metadata": {
"tags": [
"cloud/aws/ec2",
"data-storage/s3",
"workflow/randomforest",
"workflow/hpo",
"workflow/xgboost",
"library/dask",
"library/dask-cuda",
"library/xgboost",
"library/optuna",
"library/sklearn",
"library/dask-ml"
]
},
"source": [
"# HPO Benchmarking with RAPIDS and Dask\n",
"\n",
"Hyper-Parameter Optimization (HPO) helps to find the best version of a model by exploring the space of possible configurations. While generally desirable, this search is computationally expensive and time-consuming.\n",
"\n",
"In the notebook demo below, we compare benchmarking results to show how GPU can accelerate HPO tuning jobs relative to CPU.\n",
"\n",
"For instance, we find a 26x speedup in wall clock time (0.71 hrs vs. 18.9 hrs) when comparing between GPU and CPU EC2 instances on 100 XGBoost HPO trials using no parallel workers on 3 years of the Airline Dataset.\n"
"For instance, we find a 48x speedup in wall clock time (0.71 hrs vs 34.6 hrs) for XGBoost and 16x (3.86 hrs vs 63.2 hrs) for RandomForest when comparing between GPU and CPU EC2 instances on 100 HPO trials on 3 years of the Airline Dataset.\n"
]
},
{
Expand Down
Loading