[doc] Variable 'label' is not defined in the pyspark application exam…

…ple (#9302)
dmlc · Jun 15, 2023 · 2718ff5 · 2718ff5
1 parent 0df1272
commit 2718ff5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doc/tutorials/spark_estimator.rst b/doc/tutorials/spark_estimator.rst
@@ -146,7 +146,7 @@ using a list of feature names and the additional parameter ``use_gpu``:
   label_name = "class"
 
   # get a list with feature column names
-  feature_names = [x.name for x in train_df.schema if x.name != label]
+  feature_names = [x.name for x in train_df.schema if x.name != label_name]
 
   # create a xgboost pyspark regressor estimator and set use_gpu=True
   regressor = SparkXGBRegressor(