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

add basic walkthrough

Browse Source
This commit is contained in:
tqchen
2014-09-06 10:11:45 -07:00
parent 8ad9293437
commit b858283ec5
4 changed files with 111 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
R-package/R/xgboost.R
View File

@@ -40,14 +40,7 @@
#'
xgboost <- function(data = NULL, label = NULL, params = list(), nrounds,
verbose = 1, ...) {
dtrain <- xgb.get.DMatrix(data, label)
if (verbose > 1) {
silent <- 0
} else {
silent <- 1
}
params <- append(params, list(silent = silent))
dtrain <- xgb.get.DMatrix(data, label)
params <- append(params, list(...))
if (verbose > 0) {
@@ -56,7 +49,7 @@ xgboost <- function(data = NULL, label = NULL, params = list(), nrounds,
watchlist <- list()
}
bst <- xgb.train(params, dtrain, nrounds, watchlist)
bst <- xgb.train(params, dtrain, nrounds, watchlist, verbose=verbose)
return(bst)
}