diff --git a/demo/kaggle-higgs/higgs-pred.R b/demo/kaggle-higgs/higgs-pred.R index ba6d235ca..62076470d 100644 --- a/demo/kaggle-higgs/higgs-pred.R +++ b/demo/kaggle-higgs/higgs-pred.R @@ -10,7 +10,7 @@ idx <- dtest[[1]] xgmat <- xgb.DMatrix(data, missing = -999.0) bst <- xgb.Booster(params=list("nthread"=16), modelfile=modelfile) -ypred <- xgb.predict(bst, xgmat) +ypred <- predict(bst, xgmat) rorder <- rank(ypred, ties.method="first")