Train model with warnings

train_model_w_warnings(
  model_formula,
  train_data,
  method,
  cv,
  perf_metric_name,
  tune_grid,
  ntree
)

Arguments

model_formula

model formula

train_data

train_data

method

method

cv

cross-validation caret scheme

perf_metric_name

The column name from the output of the function provided to perf_metric_function that is to be used as the performance metric. Defaults: binary classification = "ROC", multi-class classification = "logLoss", regression = "RMSE".

tune_grid

tuning grid

ntree

For random forest, how many trees to use (default: 1000). Note that caret doesn't allow this parameter to be tuned.

Value

trained model

Author

Zena Lapp, zenalapp@umich.edu