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

[jvm-packages] remove APIs with DMatrix from xgboost-spark (#1519)

Browse Source 
* test consistency of prediction functions between DMatrix and RDD

* remove APIs with DMatrix from xgboost-spark

* fix compilation error in xgboost4j-example

* fix test cases
This commit is contained in:
Nan Zhu
2016-08-28 21:25:49 -04:00
committed by GitHub
parent 6d65aae091
commit 74db1e8867
3 changed files with 5 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostModel.scala
View File

@@ -122,15 +122,6 @@ class XGBoostModel(_booster: Booster) extends Serializable {
}
}
/**
* Predict result with the given test set (represented as DMatrix)
*
* @param testSet test set represented as DMatrix
*/
def predict(testSet: DMatrix): Array[Array[Float]] = {
_booster.predict(testSet)
}
/**
* Predict leaf instances with the given test set (represented as RDD)
*
@@ -149,15 +140,6 @@ class XGBoostModel(_booster: Booster) extends Serializable {
}
}
/**
* Predict leaf instances with the given test set (represented as DMatrix)
*
* @param testSet test set represented as DMatrix
*/
def predictLeaves(testSet: DMatrix): Array[Array[Float]] = {
_booster.predictLeaf(testSet, 0)
}
/**
* Save the model as to HDFS-compatible file system.
*