wrapper_xgboost: Extreme Gradient Boosting wrapper for CCI

View source: R/wrappers.R

wrapper_xgboostR Documentation

Extreme Gradient Boosting wrapper for CCI

Description

Extreme Gradient Boosting wrapper for CCI

Usage

wrapper_xgboost(
  formula,
  data,
  train_indices,
  test_indices,
  metric,
  nrounds = 500,
  metricfunc = NULL,
  nthread = 1,
  num_class = NULL,
  subsample = 1,
  ...
)

Arguments

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

Value

Performance metric


CCI documentation built on Aug. 29, 2025, 5:17 p.m.