Entrena un árbol de regresión
Como ya sabes, los árboles de decisión son una herramienta útil para problemas de clasificación. Además, también puedes usarlos para modelar problemas de regresión. La diferencia estructural es que en los nodos hoja habrá valores numéricos (en lugar de clases).
En este ejercicio, usarás el conjunto de datos de chocolate para ajustar un árbol de regresión. Esto es muy similar a lo que hiciste en el Capítulo 1 con el conjunto diabetes.
En tu espacio de trabajo tienes disponibles los datos de entrenamiento chocolate_train.
Este ejercicio forma parte del curso
Machine Learning con modelos basados en árboles en R
Instrucciones del ejercicio
- Crea
model_spec, una especificación de árbol de regresión. - Usando el data frame
chocolate_train, utilizamodel_specpara entrenar un árbol de regresión que predigafinal_gradeusando solo los predictores numéricos de los datos.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
library(tidymodels)
# Build the specification
model_spec <- decision_tree() %>%
set_mode(___) %>%
set_engine(___)
# Fit to the data
model_fit <- model_spec %>%
___(formula = ___,
data = ___)
model_fit