[jvm-packages] test consistency of prediction functions with DMatrix and RDD (#1518)

* test consistency of prediction functions between DMatrix and RDD * fix the failed test cases
2016-08-28 20:27:03 -04:00
parent d7f79255ec
commit 6d65aae091
2 changed files with 105 additions and 84 deletions
--- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostModel.scala
+++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostModel.scala
@@ -128,7 +128,7 @@ class XGBoostModel(_booster: Booster) extends Serializable {
   * @param testSet test set represented as DMatrix
   */
  def predict(testSet: DMatrix): Array[Array[Float]] = {
-    _booster.predict(testSet, true, 0)
+    _booster.predict(testSet)
  }

  /**