[jvm-packages] adding logs for parameters (#4091)

This commit is contained in:
Nan Zhu
2019-01-30 21:50:55 -08:00
committed by GitHub
parent a60e224484
commit 0d0ce32908

View File

@@ -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)