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

fix bug in format of input

Browse Source
This commit is contained in:
hetong 2015-01-19 19:26:25 -08:00
parent f295177b1d
commit c0c6951b73
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
R-package/R/predict.xgb.Booster.R
View File

@ -56,7 +56,7 @@ setMethod("predict", signature = "xgb.Booster",
if (length(ret) == len){ if (length(ret) == len){
ret <- matrix(ret,ncol = 1) ret <- matrix(ret,ncol = 1)
} else { } else {
ret <- matrix(ret, ncol = nrow(newdata)) ret <- matrix(ret, ncol = len)
ret <- t(ret) ret <- t(ret)
} }
} }

1
R-package/demo/predict_leaf_indices.R
View File

@ -12,7 +12,6 @@ nround = 5
# training the model for two rounds # training the model for two rounds
bst = xgb.train(param, dtrain, nround, watchlist) bst = xgb.train(param, dtrain, nround, watchlist)
cat('start testing prediction from first n trees\n') cat('start testing prediction from first n trees\n')
labels <- getinfo(dtest,'label')
### predict using first 2 tree ### predict using first 2 tree
pred_with_leaf = predict(bst, dtest, ntreelimit = 2, predleaf = TRUE) pred_with_leaf = predict(bst, dtest, ntreelimit = 2, predleaf = TRUE)