From 5d093a7f4cd2e7f2a52ecbb3125791eb97dbdf95 Mon Sep 17 00:00:00 2001 From: moqiguzhu Date: Sat, 25 Feb 2017 06:16:19 -0800 Subject: [PATCH] in `caret` settings, if you want do 10*10 cross validation, you need to set repeats=10, number=10 and method=repeatedcv, (#2061) if you set method=cv, actually just one 10-fold cross validation will be run; fixes #2055 --- R-package/demo/caret_wrapper.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R-package/demo/caret_wrapper.R b/R-package/demo/caret_wrapper.R index 13e05e562..751b202b5 100644 --- a/R-package/demo/caret_wrapper.R +++ b/R-package/demo/caret_wrapper.R @@ -24,7 +24,7 @@ df[,ID:=NULL] #-------------Basic Training using XGBoost in caret Library----------------- # Set up control parameters for caret::train # Here we use 10-fold cross-validation, repeating twice, and using random search for tuning hyper-parameters. -fitControl <- trainControl(method = "cv", number = 10, repeats = 2, search = "random") +fitControl <- trainControl(method = "repeatedcv", number = 10, repeats = 2, search = "random") # train a xgbTree model using caret::train model <- train(factor(Improved)~., data = df, method = "xgbTree", trControl = fitControl)