[doc] Variable 'label' is not defined in the pyspark application example (#9302)

2023-06-15 18:06:52 -03:00
parent 0df1272695
commit 2718ff530c
1 changed files with 1 additions and 1 deletions
--- 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(