Ajusta un random forest con una optimización personalizada
Ahora que has explorado las cuadrículas de optimización predeterminadas que ofrece train(), vamos a personalizar un poco más tus modelos.
Puedes proporcionar cualquier número de valores para mtry, desde 2 hasta el número de columnas del conjunto de datos. En la práctica, el rendimiento apenas mejora con valores muy grandes de mtry, así que usarás una cuadrícula personalizada que pruebe 2 modelos sencillos (mtry = 2 y mtry = 3) y uno más complejo (mtry = 7).
Este ejercicio forma parte del curso
Machine Learning con caret en R
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Define the tuning grid: tuneGrid
tuneGrid <- data.frame(
.mtry = ___,
.splitrule = "___",
.min.node.size = ___
)