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

fix some wording

Browse Source
This commit is contained in:
Vadim Khotilovich 2015-04-02 19:59:06 -05:00
parent b8711226e2
commit 611d69c771
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -222,10 +222,10 @@ xgb.cv.mknfold <- function(dall, nfold, param, stratified) {
y <- getinfo(dall, 'label')
if (stratified & length(y) == length(randidx)) {
y <- y[randidx]
# by default assume that y is a classification label, and only
# leave it numeric for the reg:linear objective
# WARNING: if there would be any objectives with truly numerical
# they would not currently be treated correctly.
# By default assume that y is a classification label,
# and only leave it numeric for the reg:linear objective.
# WARNING: if there would be any other objectives with truly
# numerical labels, they currently would not be treated correctly!
if (param[['objective']] != 'reg:linear') y <- factor(y)
idset <- xgb.createFolds(y, nfold)
} else {