diff --git a/examples/notebooks/python/mortgage-gpu.ipynb b/examples/notebooks/python/mortgage-gpu.ipynb index e117cd56f51..e05f82fb8f7 100644 --- a/examples/notebooks/python/mortgage-gpu.ipynb +++ b/examples/notebooks/python/mortgage-gpu.ipynb @@ -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" ] },