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

check whether objective is character

Browse Source
This commit is contained in:
Tong He 2015-06-11 14:04:43 -07:00
parent fbaa3821a4
commit 61142f203b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
R-package/R/utils.R
View File

@ -220,7 +220,8 @@ xgb.cv.mknfold <- function(dall, nfold, param, stratified, folds) {
stop("nfold must be bigger than 1") stop("nfold must be bigger than 1")
} }
if(is.null(folds)) { if(is.null(folds)) {
if (exists('objective', where=param) && strtrim(param[['objective']], 5) == 'rank:') { if (exists('objective', where=param) && is.character(param$objective) &&
strtrim(param[['objective']], 5) == 'rank:') {
stop("\tAutomatic creation of CV-folds is not implemented for ranking!\n", stop("\tAutomatic creation of CV-folds is not implemented for ranking!\n",
"\tConsider providing pre-computed CV-folds through the folds parameter.") "\tConsider providing pre-computed CV-folds through the folds parameter.")
} }
@ -234,7 +235,7 @@ 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.
if (exists('objective', where=param)) { if (exists('objective', where=param) && is.character(param$objective)) {
# 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
if (param[['objective']] != 'reg:linear') y <- factor(y) if (param[['objective']] != 'reg:linear') y <- factor(y)