XGBoost Learner

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

Usage

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

Arguments

data

A dataframe to train a learner / learners on.

formula

A regression formula to use inside this learner.

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