tests/testthat/setup-tornado_train.R

In bertcarnell/tornado: Plots for Model Sensitivity and Variable Importance

# Copyright 2022 Robert Carnell

if (requireNamespace("caret", quietly = TRUE)) {
  rf_model_numeric <- caret::train(x = subset(mtcars, select = -mpg), y = mtcars$mpg, method = "rf")
  rf_model_numeric_weights <- caret::train(x = subset(mtcars, select = -mpg), y = mtcars$mpg, method = "rf", weights = 1:nrow(mtcars))

  if (!exists("my_mtcars"))
  {
    source("setup-create_endpoints.R")
  }

  rf_model_factor <- caret::train(x = subset(my_mtcars, select = -mpg), y = my_mtcars$mpg, method = "rf")

  rf_model_class <- caret::train(x = subset(my_mtcars, select = -vs), y = my_mtcars$vs, method = "rf")

  stepAIC_model_numeric <- caret::train(x = subset(mtcars, select = -mpg),
                                        y = mtcars$mpg, method = "glmStepAIC", trace = 0)
}