Fix invalid param value "use_gpu":"true"

After installing latest Jupyter notebooke, the json format check is strick, due to passing a string value "true" for the "use_gpu":"true", "use_gpu": True is required. ``` params = { "tree_method": "gpu_hist", "grow_policy": "depthwise", "num_workers": 1, "use_gpu": "true", } ``` TypeError: Invalid param value given for param "use_gpu". Boolean Param requires value of type bool. Found <class 'str'>. More over, "use_gpu": True is deprecated in dmlc/XGBoost, change to user "device": "cuda" instead Signed-off-by: Tim Liu <[email protected]>
NVIDIA · May 21, 2024 · 3e5fd9b · 3e5fd9b
1 parent 6436748
commit 3e5fd9b
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb b/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb
@@ -197,7 +197,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",

diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb
@@ -933,7 +933,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",

diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb
@@ -191,7 +191,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "\n",
     "params['features_col'] = features\n",

diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb
@@ -235,7 +235,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",

diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb
@@ -182,7 +182,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",

diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb
@@ -213,7 +213,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",