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

Update xgb.cv.R

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

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

@ -124,14 +124,14 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
if (!is.null(params$objective)) if (!is.null(params$objective))
if (class(params$objective)=='function') { if (class(params$objective)=='function') {
obj = params$objective obj = params$objective
params$objective = NULL params[['objective']] = NULL
} }
if (!is.null(params$eval_metric) && !is.null(feval)) if (!is.null(params$eval_metric) && !is.null(feval))
stop("xgb.cv: cannot assign two different evaluation metrics") stop("xgb.cv: cannot assign two different evaluation metrics")
if (!is.null(params$eval_metric)) if (!is.null(params$eval_metric))
if (class(params$eval_metric)=='function') { if (class(params$eval_metric)=='function') {
feval = params$eval_metric feval = params$eval_metric
params$eval_metric = NULL params[['eval_metric']] = NULL
} }
# Early Stopping # Early Stopping