wrapper_xgboost | R Documentation |
Extreme Gradient Boosting wrapper for CCI
wrapper_xgboost(
formula,
data,
train_indices,
test_indices,
metric,
nrounds = 500,
metricfunc = NULL,
nthread = 1,
num_class = NULL,
subsample = 1,
...
)
formula |
Model formula |
data |
Data frame |
train_indices |
Indices for training data |
test_indices |
Indices for training data |
metric |
Type of performance metric |
nrounds |
Number of boosting rounds |
metricfunc |
A user specific metric function which have the arguments data, model test_indices and test_matrix and returns a numeric value |
nthread |
Integer. Number of threads to use for parallel computation during model training in XGBoost. Default is 1. |
num_class |
Number of categorical classes |
subsample |
Proportion of the data to be used. Default is 1 (no subsampling). |
... |
Additional arguments passed to xgb.train |
Performance metric
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.