predict.bagger {baguette}

Predictions from a bagged model

Description

The predict() function computes predictions from each of the models in the ensembles and returns a single aggregated value for each sample in new_data.

Usage

## S3 method for class 'bagger'
predict(object, new_data, type = NULL, ...)

Arguments

object

An object generated by bagger().

new_data

A data frame of predictors. If a recipe or formula were originally used, the original data should be passed here instead of a preprocessed version.

type

A single character value for the type of predictions. For regression models, type = 'numeric' is valid and 'class' and 'prob' are valid for classification models.

...

Not currently used.

Examples

data(airquality)

set.seed(7687)
cart_bag <- bagger(Ozone ~ ., data = airquality, base_model = "CART", times = 5)
predict(cart_bag, new_data = airquality[, -1])