ÎncepețiÎncepe gratuit

Antrenează un arbore de regresie

Știi deja că arborii de decizie sunt un instrument util pentru problemele de clasificare. Ei pot fi folosiți însă și pentru modelarea problemelor de regresie. Diferența structurală constă în faptul că nodurile frunză vor conține valori numerice în loc de clase.

În acest exercițiu, vei folosi setul de date despre ciocolată pentru a antrena un arbore de regresie. Procesul este foarte similar cu ce ai făcut deja în Capitolul 1 cu setul de date diabetes.

În spațiul de lucru ai la dispoziție datele de antrenament chocolate_train.

Acest exercițiu face parte din cursul

Machine Learning cu modele bazate pe arbori în R

Vezi cursul

Instrucțiuni pentru exercițiu

  • Construiește model_spec, o specificație de arbore de regresie.
  • Folosind cadrul de date chocolate_train, utilizează model_spec pentru a antrena un arbore de regresie care să prezică final_grade folosind doar predictorii numerici din date.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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
Editează și rulează codul