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

fix multi cv pred

Browse Source
This commit is contained in:
unknown 2015-03-29 00:02:29 -07:00
parent 37567e440c
commit 431277d5ca
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
R-package/R/xgb.cv.R
View File

@ -121,7 +121,8 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
} else { } else {
res <- xgb.iter.eval(fd$booster, fd$watchlist, i - 1, feval, prediction) res <- xgb.iter.eval(fd$booster, fd$watchlist, i - 1, feval, prediction)
if (mat_pred) { if (mat_pred) {
predictValues[fd$index,] <- res[[2]] pred_mat = matrix(res[[2]],num_class,length(fd$index))
predictValues[fd$index,] <- t(pred_mat)
} else { } else {
predictValues[fd$index] <- res[[2]] predictValues[fd$index] <- res[[2]]
} }