From 7cb449c4a75c2a16f6dfea5244ce959d998344b1 Mon Sep 17 00:00:00 2001 From: Tong He Date: Thu, 11 Jun 2015 14:16:20 -0700 Subject: [PATCH] Update xgb.cv.R --- R-package/R/xgb.cv.R | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/R-package/R/xgb.cv.R b/R-package/R/xgb.cv.R index df7fd5648..e7687ac3d 100644 --- a/R-package/R/xgb.cv.R +++ b/R-package/R/xgb.cv.R @@ -124,14 +124,14 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing = if (!is.null(params$objective)) if (class(params$objective)=='function') { obj = params$objective - params$objective = NULL + params[['objective']] = NULL } if (!is.null(params$eval_metric) && !is.null(feval)) stop("xgb.cv: cannot assign two different evaluation metrics") if (!is.null(params$eval_metric)) if (class(params$eval_metric)=='function') { feval = params$eval_metric - params$eval_metric = NULL + params[['eval_metric']] = NULL } # Early Stopping