R/constructor.R

In baguette: Efficient Model Functions for Bagging

new_bagger <- function(model_df, imp, control, cost, base_model, blueprint,
                       call = rlang::caller_env()) {

  if (!is_tibble(model_df)) {
    cli::cli_abort("{.arg model_df} should be a {.cls tibble}, not
                   {obj_type_friendly(model_df)}.", call = call)
  }

  # TODO extend to use mode from model object(s)
  if (is.numeric(blueprint$ptypes$outcomes[[1]])) {
    mod_mode <- "regression"
  } else {
    mod_mode <- "classification"
  }

  hardhat::new_model(
    model_df = model_df,
    control = control,
    cost = cost,
    imp = imp,
    base_model = c(base_model[1], mod_mode),
    blueprint = blueprint,
    class = "bagger"
  )
}