2025-02-03
This is a presentation that features a Plotly plot.
data <- mtcars
data$am <- as.factor(data$am)
data$vs <- as.factor(data$vs)
set.seed(123)
y <- data$mpg
x <- data[, !(names(mtcars) %in% c("mpg"))]
control <- rfeControl(functions = rfFuncs,
method = "cv",
number = 10)
results <- rfe(x, y,
sizes = c(1:10),
rfeControl = control)
if (length(results$optVariables) < 4) {
selected_variables <- results$optVariables
} else {
selected_variables <- results$optVariables[1:4]
}
print(selected_variables)## [1] "disp" "wt" "hp" "cyl"