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 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])
#> # A tibble: 153 × 1
#> .pred
#> <dbl>
#> 1 29.9
#> 2 20.5
#> 3 17.0
#> 4 22.3
#> 5 14.0
#> 6 18.7
#> 7 23.6
#> 8 17.4
#> 9 12.2
#> 10 23.5
#> # ℹ 143 more rows