ComeçarComece de graça

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

Ver curso

Instruções do exercício

  • Construa model_spec, uma especificação de árvore de regressão.
  • Usando o data frame chocolate_train, use model_spec para treinar uma árvore de regressão que preveja final_grade usando 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
Editar e executar o código