xgboost/R-package/inst/examples/cross_validation.R
2014-09-05 22:46:09 -07:00

11 lines
285 B
R

require(xgboost)
require(methods)
# Directly read in local file
dtrain <- xgb.DMatrix("agaricus.txt.train")
history <- xgb.cv( data = dtrain, nround=3, nfold = 5, metrics=list("rmse","auc"),
"max_depth"=3, "eta"=1,
"objective"="binary:logistic")