From 0d0ce32908b4081e9fe85212177af6ae01564f4d Mon Sep 17 00:00:00 2001 From: Nan Zhu Date: Wed, 30 Jan 2019 21:50:55 -0800 Subject: [PATCH] [jvm-packages] adding logs for parameters (#4091) --- .../src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala | 1 + 1 file changed, 1 insertion(+) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala index 4fe82b271..7914fb817 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala @@ -366,6 +366,7 @@ object XGBoost extends Serializable { hasGroup: Boolean = false, evalSetsMap: Map[String, RDD[XGBLabeledPoint]] = Map()): (Booster, Map[String, Array[Float]]) = { + logger.info(s"XGBoost training with parameters:\n${params.mkString("\n")}") val (nWorkers, round, _, _, _, _, trackerConf, timeoutRequestWorkers, checkpointPath, checkpointInterval) = parameterFetchAndValidation(params, trainingData.sparkContext)