Fit the best model from each algorithm to the data.

Fit the best model from each algorithm to the data.

fit_best_model(
  lasso_wf,
  best_lasso,
  rand_wf,
  best_rand,
  xg_wf,
  best_xg,
  train_x_class,
  train_y_class,
  category
)

Arguments

lasso_wf

A lasso workflow (including the search space for the model)

best_lasso

The best model output for lasso

rand_wf

A random forest workflow (including the search space for the model)

best_rand

The best model output for random forest

xg_wf

An XGBoost workflow (including the search space for the model)

best_xg

The best model output for XGBoost

train_x_class

Training data for predictors

train_y_class

Training data for outcomes

category

The target binary category.

Value

A list output that contains the best output for lasso, random forest, and XGBoost.