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

[jvm-packages] Comply with scala style convention + fix broken unit test (#5134)

Browse Source 
* Fix scala style check

* fix messed unit test
This commit is contained in:
Philip Hyunsu Cho
2019-12-18 17:26:58 -08:00
committed by GitHub
parent bc9d88259f
commit 37fdfa03f8
5 changed files with 27 additions and 23 deletions
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/MissingValueHandlingSuite.scala
View File

@@ -151,7 +151,7 @@ class MissingValueHandlingSuite extends FunSuite with PerTest {
}
}
test("specify a non-zero missing value but set allow_non_zero_missing_value " +
test("specify a non-zero missing value but set allow_non_zero_missing " +
"does not stop application") {
val spark = ss
import spark.implicits._
@@ -174,7 +174,7 @@ class MissingValueHandlingSuite extends FunSuite with PerTest {
inputDF.show()
val paramMap = List("eta" -> "1", "max_depth" -> "2",
"objective" -> "binary:logistic", "missing" -> -1.0f,
"num_workers" -> 1, "allow_non_zero_for_missing_value" -> "true").toMap
"num_workers" -> 1, "allow_non_zero_for_missing" -> "true").toMap
val model = new XGBoostClassifier(paramMap).fit(inputDF)
model.transform(inputDF).collect()
}