XGBoost Learner — lnr_xgboost • nadir

A wrapper for xgboost::xgboost() for use in nadir::super_learner().

Usage

lnr_xgboost(data, formula, weights = NULL, nrounds = 1000, verbose = 0, ...)

Arguments

data: A dataframe to train a learner / learners on.
formula: A regression formula to use inside this learner.
weights: Observation weights; see ?lm
nrounds: The max number of boosting iterations
verbose: If verbose is > 0 then xgboost::xgboost() will print out messages about its fitting process. See ?xgboost::xgboost
...: Any extra arguments that should be passed to the internal model for model fitting purposes.

Value

A prediction function that accepts newdata, which returns predictions (a numeric vector of values, one for each row of newdata).

See also

learners