Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[jvm-packages] Fixed CheckpointManagerSuite for Scala 2.10 (#3332)

Browse Source 
As before, the compilation error is caused by mixing positional and
labelled arguments.
This commit is contained in:
Sergei Lebedev 2018-05-20 03:28:11 +02:00 committed by Nan Zhu
parent 3ee725e3bb
commit 8f6aadd4b7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/CheckpointManagerSuite.scala
View File

@ -38,8 +38,8 @@ class CheckpointManagerSuite extends FunSuite with BeforeAndAfterAll {
val trainingRDD = sc.parallelize(Classification.train).map(_.asML).cache() val trainingRDD = sc.parallelize(Classification.train).map(_.asML).cache()
val paramMap = Map("eta" -> "1", "max_depth" -> "2", "silent" -> "1", val paramMap = Map("eta" -> "1", "max_depth" -> "2", "silent" -> "1",
"objective" -> "binary:logistic") "objective" -> "binary:logistic")
(XGBoost.trainWithRDD(trainingRDD, paramMap, round = 2, sc.defaultParallelism), (XGBoost.trainWithRDD(trainingRDD, paramMap, round = 2, nWorkers = sc.defaultParallelism),
XGBoost.trainWithRDD(trainingRDD, paramMap, round = 4, sc.defaultParallelism)) XGBoost.trainWithRDD(trainingRDD, paramMap, round = 4, nWorkers = sc.defaultParallelism))
} }
test("test update/load models") { test("test update/load models") {