xgb_train  R Documentation 
xgb_train()
and xgb_predict()
are wrappers for xgboost
treebased
models where all of the model arguments are in the main function.
xgb_train(
x,
y,
weights = NULL,
max_depth = 6,
nrounds = 15,
eta = 0.3,
colsample_bynode = NULL,
colsample_bytree = NULL,
min_child_weight = 1,
gamma = 0,
subsample = 1,
validation = 0,
early_stop = NULL,
counts = TRUE,
event_level = c("first", "second"),
...
)
xgb_predict(object, new_data, ...)
x 
A data frame or matrix of predictors 
y 
A vector (factor or numeric) or matrix (numeric) of outcome data. 
max_depth 
An integer for the maximum depth of the tree. 
nrounds 
An integer for the number of boosting iterations. 
eta 
A numeric value between zero and one to control the learning rate. 
colsample_bynode 
Subsampling proportion of columns for each node
within each tree. See the 
colsample_bytree 
Subsampling proportion of columns for each tree.
See the 
min_child_weight 
A numeric value for the minimum sum of instance weights needed in a child to continue to split. 
gamma 
A number for the minimum loss reduction required to make a further partition on a leaf node of the tree 
subsample 
Subsampling proportion of rows. By default, all of the training data are used. 
validation 
The proportion of the data that are used for performance assessment and potential early stopping. 
early_stop 
An integer or 
counts 
A logical. If 
event_level 
For binary classification, this is a single string of either

... 
Other options to pass to 
new_data 
A rectangular data object, such as a data frame. 
A fitted xgboost
object.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.