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

[jvm-packages] predictLeaf with Dataframe (#1576)

Browse Source 
* add back train method but mark as deprecated

* predictLeaf with Dataset

* fix

* fix
This commit is contained in:
Nan Zhu
2016-09-15 06:15:47 -04:00
committed by GitHub
parent bb388cbb31
commit 4ad648e856
3 changed files with 68 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostGeneralSuite.scala
View File

@@ -133,13 +133,12 @@ class XGBoostGeneralSuite extends SharedSparkContext with Utils {
}
test("test eval functions with RDD") {
val trainingRDD = buildTrainingRDD(sc)
val trainingRDD = buildTrainingRDD(sc).cache()
val paramMap = List("eta" -> "1", "max_depth" -> "2", "silent" -> "0",
"objective" -> "binary:logistic").toMap
val xgBoostModel = XGBoost.trainWithRDD(trainingRDD, paramMap, 5, numWorkers)
val evalFunc = new EvalError
xgBoostModel.eval(trainingRDD, "eval1", iter = 5, useExternalCache = false)
xgBoostModel.eval(trainingRDD, "eval2", evalFunc = evalFunc, useExternalCache = false)
xgBoostModel.eval(trainingRDD, "eval2", evalFunc = new EvalError, useExternalCache = false)
}
test("test prediction functionality with empty partition") {