add training log on the driver side

wbo4958 · Aug 9, 2023 · ba75358 · ba75358
1 parent c1b2cff
commit ba75358
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/python-package/xgboost/spark/core.py b/python-package/xgboost/spark/core.py
@@ -995,7 +995,18 @@ def _run_job() -> Tuple[str, str]:
             )
             return ret[0], ret[1]
 
+        get_logger("xgboost-pyspark").info(
+            "Running XGBoostPySpark-{%s} with "
+            "booster params: %s\n"
+            "train_call_kwargs_params: %s\n"
+            "dmatrix_kwargs: %s",
+            xgboost._py_version(),
+            booster_params,
+            train_call_kwargs_params,
+            dmatrix_kwargs)
         (config, booster) = _run_job()
+        get_logger("xgboost-pyspark").info("Finished XGBoostPySpark training")
+
 
         result_xgb_model = self._convert_to_sklearn_model(
             bytearray(booster, "utf-8"), config