Update understandingXGBoostModel.Rmd

viviancodes · May 4, 2015 · a3ad9df · a3ad9df
1 parent 2157146
commit a3ad9df
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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,