fit_best_model: Fit the best model from each algorithm to the data.

In snfagora/autotextclassifier: R Package for Automatically Classifying Texts Based on Tidymodels

Fit the best model from each algorithm to the data.

Description

Fit the best model from each algorithm to the data.

Usage

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.

snfagora/autotextclassifier documentation built on June 16, 2022, 9:42 a.m.