Skip to content

Commit

Permalink
Minor fixes to xgboost example (#2832)
Browse files Browse the repository at this point in the history
Co-authored-by: Yuan-Ting Hsieh (謝沅廷) <[email protected]>
Co-authored-by: Ziyue Xu <[email protected]>
  • Loading branch information
3 people authored Aug 23, 2024
1 parent 7ab37e1 commit 141c8bd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions examples/advanced/xgboost/data_job_setup.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@
"outputs": [],
"source": [
"# please change this DATASET_ROOT to the correct path containing HIGGS dataset\n",
"%env DATASET_ROOT=/data\n",
"%env DATASET_ROOT=$HOME/dataset\n",
"!python3 utils/prepare_data_split.py \\\n",
" --data_path \"${DATASET_ROOT}/HIGGS.csv\" \\\n",
" --site_num 5 \\\n",
Expand Down Expand Up @@ -96,15 +96,15 @@
"%env DATA_SPLIT_ROOT=/tmp/nvflare/xgboost/HIGGS/data_splits/\n",
"!python3 utils/prepare_job_config.py \\\n",
" --site_num 5 \\\n",
" --training_mode bagging \\\n",
" --training_algo bagging \\\n",
" --split_method uniform \\\n",
" --lr_mode uniform \\\n",
" --nthread 16 \\\n",
" --tree_method \"hist\" \\\n",
" --data_root \"${DATA_SPLIT_ROOT}\"\n",
"!python3 utils/prepare_job_config.py \\\n",
" --site_num 5 \\\n",
" --training_mode histogram \\\n",
" --training_algo histogram \\\n",
" --split_method uniform \\\n",
" --lr_mode uniform \\\n",
" --nthread 16 \\\n",
Expand Down

0 comments on commit 141c8bd

Please sign in to comment.