From a3ad9df0b4d330dddc3901c444092cd32932d071 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Mon, 4 May 2015 14:27:44 -0700 Subject: [PATCH] Update understandingXGBoostModel.Rmd --- demo/kaggle-otto/understandingXGBoostModel.Rmd | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/demo/kaggle-otto/understandingXGBoostModel.Rmd b/demo/kaggle-otto/understandingXGBoostModel.Rmd index e626147a595c..6bd670c82a30 100644 --- a/demo/kaggle-otto/understandingXGBoostModel.Rmd +++ b/demo/kaggle-otto/understandingXGBoostModel.Rmd @@ -111,13 +111,13 @@ Look at the function documentation for more information. ```{r crossValidation} -numberOfClasses <- max(y) +numberOfClasses <- max(y) + 1 param <- list("objective" = "multi:softprob", "eval_metric" = "mlogloss", - "num_class" = numberOfClasses + 1) + "num_class" = numberOfClasses) -cv.nround <- 50 +cv.nround <- 5 cv.nfold <- 3 bst.cv = xgb.cv(param=param, data = trainMatrix, label = y,