Skip to content

Commit

Permalink
Update mortgage-gpu.ipynb (NVIDIA#72)
Browse files Browse the repository at this point in the history
  • Loading branch information
sameerz authored and krajendrannv committed Jan 23, 2020
1 parent da110fd commit 67d3222
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions examples/notebooks/python/mortgage-gpu.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,10 @@
" 1. NumPy\n",
" 2. cudf jar\n",
" 2. xgboost4j jar\n",
" 3. xgboost4j-spark jar"

"This notebook illustrates the ease of porting a sample CPU based Spark xgboost4j code into GPU. There are two changes required for running Spark XGBoost on GPU:\n"
" 1. Data reader uses `GpuDataReader(spark)` instead of `spark.read`"
" 2. `setFeaturesCol` function on CPU is replaced by setFeaturesCols(features)"
" 3. xgboost4j-spark jar",
"This notebook illustrates the ease of porting a sample CPU based Spark xgboost4j code into GPU. There are two changes required for running Spark XGBoost on GPU:\n",
" 1. Data reader uses `GpuDataReader(spark)` instead of `spark.read`",
" 2. `setFeaturesCol` function on CPU is replaced by setFeaturesCols(features)",
" a. This also eliminates the need for vectorization (assembling multiple feature columns in to one column) since we can read multiple columns"
]
},
Expand Down

0 comments on commit 67d3222

Please sign in to comment.