hide xgb.Boost

2014-08-27 22:25:54 -07:00
parent 8a4e66299a
commit a0f22f6aaa
2 changed files with 1 additions and 2 deletions
--- a/demo/kaggle-higgs/higgs-pred.R
+++ b/demo/kaggle-higgs/higgs-pred.R
@@ -9,7 +9,7 @@ data <- as.matrix(dtest[2:31])
 idx <- dtest[[1]]

 xgmat <- xgb.DMatrix(data, missing = -999.0)
-bst <- xgb.Booster(params=list("nthread"=16), modelfile=modelfile)
+bst <- xgb.load(modelfile=modelfile)
 ypred <- predict(bst, xgmat)

 rorder <- rank(ypred, ties.method="first")