This commit is contained in:
Vadim Khotilovich 2015-05-01 16:16:50 -05:00
parent f05c7d87cb
commit c18e081f48

View File

@ -234,7 +234,6 @@ xgb.cv.mknfold <- function(dall, nfold, param, stratified, folds) {
# For classification, need to convert y labels to factor before making the folds, # For classification, need to convert y labels to factor before making the folds,
# and then do stratification by factor levels. # and then do stratification by factor levels.
# For regression, leave y numeric and do stratification by quantiles. # For regression, leave y numeric and do stratification by quantiles.
n_uniq <- length(unique(y))
if (exists('objective', where=param)) { if (exists('objective', where=param)) {
# If 'objective' provided in params, assume that y is a classification label # If 'objective' provided in params, assume that y is a classification label
# unless objective is reg:linear # unless objective is reg:linear