Skip to content

Commit

Permalink
remove APIs with DMatrix from xgboost-spark
Browse files Browse the repository at this point in the history
  • Loading branch information
CodingCat committed Aug 29, 2016
1 parent e58ed7f commit 406004a
Showing 1 changed file with 0 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -140,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.
*
Expand Down

0 comments on commit 406004a

Please sign in to comment.