Treine uma árvore de regressão
Como você já sabe, árvores de decisão são uma ferramenta útil para problemas de classificação. Além disso, você também pode usá-las para modelar problemas de regressão. A diferença estrutural é que haverá valores numéricos (em vez de classes) nos nós folha.
Neste exercício, você vai usar o conjunto de dados de chocolate para ajustar uma árvore de regressão. Isso é muito parecido com o que você já fez no Capítulo 1 com o conjunto diabetes.
Os dados de treino chocolate_train estão disponíveis no seu ambiente.
Este exercício faz parte do curso
Machine Learning com modelos baseados em árvores em R
Instruções do exercício
- Construa
model_spec, uma especificação de árvore de regressão. - Usando o data frame
chocolate_train, usemodel_specpara treinar uma árvore de regressão que prevejafinal_gradeusando apenas os preditores numéricos dos dados.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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