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

Update understandingXGBoostModel.Rmd

Browse Source
This commit is contained in:
Tianqi Chen 2015-05-04 14:27:44 -07:00
parent 2157146cea
commit a3ad9df0b4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/kaggle-otto/understandingXGBoostModel.Rmd
View File

@ -111,13 +111,13 @@ Look at the function documentation for more information.
```{r crossValidation} ```{r crossValidation}
numberOfClasses <- max(y) numberOfClasses <- max(y) + 1
param <- list("objective" = "multi:softprob", param <- list("objective" = "multi:softprob",
"eval_metric" = "mlogloss", "eval_metric" = "mlogloss",
"num_class" = numberOfClasses + 1) "num_class" = numberOfClasses)
cv.nround <- 50 cv.nround <- 5
cv.nfold <- 3 cv.nfold <- 3
bst.cv = xgb.cv(param=param, data = trainMatrix, label = y, bst.cv = xgb.cv(param=param, data = trainMatrix, label = y,