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

print.xgb.cv fix

Browse Source
This commit is contained in:
Vadim Khotilovich 2016-06-09 09:29:19 -05:00
parent f34f9fb9f7
commit 79704cdfb4
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -234,7 +234,7 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
#' @examples #' @examples
#' data(agaricus.train, package='xgboost') #' data(agaricus.train, package='xgboost')
#' train <- agaricus.train #' train <- agaricus.train
#' cv <- xgbcv(data = train$data, label = train$label, max.depth = 2, #' cv <- xgb.cv(data = train$data, label = train$label, nfold = 5, max.depth = 2,
#' eta = 1, nthread = 2, nround = 2, objective = "binary:logistic") #' eta = 1, nthread = 2, nround = 2, objective = "binary:logistic")
#' print(cv) #' print(cv)
#' print(cv, verbose=TRUE) #' print(cv, verbose=TRUE)
@ -269,8 +269,7 @@ print.xgb.cv.synchronous <- function(x, verbose=FALSE, ...) {
next next
cat(n, ': ', x[[n]], '\n', sep='') cat(n, ': ', x[[n]], '\n', sep='')
} }
cat('nfolds: ', length(x$folds), '\n', sep='')
if (!is.null(x$pred)) { if (!is.null(x$pred)) {
cat('pred:\n') cat('pred:\n')
str(x$pred) str(x$pred)