We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9f7b94c commit 729fd97Copy full SHA for 729fd97
doc/tutorials/spark_estimator.rst
@@ -28,7 +28,7 @@ We can create a ``SparkXGBRegressor`` estimator like:
28
.. code-block:: python
29
30
from xgboost.spark import SparkXGBRegressor
31
- spark_reg_estimator = SparkXGBRegressor(
+ xgb_regressor = SparkXGBRegressor(
32
features_col="features",
33
label_col="label",
34
num_workers=2,
@@ -61,7 +61,7 @@ type or spark array type.
61
62
63
64
- transformed_test_spark_dataframe = xgb_regressor.predict(test_spark_dataframe)
+ transformed_test_spark_dataframe = xgb_regressor_model.transform(test_spark_dataframe)
65
66
67
The above snippet code returns a ``transformed_test_spark_dataframe`` that contains the input
0 commit comments