selectBestModel()

Selects best model by F1 score, AUPRC, MCC, or balanced accuracy.

selectBestModel(tune_res, wflow, select_best_metric = "mcc")

Arguments

tune_res

Results of grid tuning, such as the output of tuneGrid()

wflow

A workflows object, such as the output of buildWflow()

select_best_metric

rlang::chr Metric to select best model: "f_meas", "pr_auc", "mcc", or "bal_accuracy"

Value

Best model workflow