Aan de slagGa gratis aan de slag

Train een regressieboom

Zoals je al weet, zijn beslisbomen een handig hulpmiddel voor classificatieproblemen. Daarnaast kun je ze ook gebruiken om regressieproblemen te modelleren. Het structurele verschil is dat er numerieke waarden (in plaats van klassen) op de bladknopen staan.

In deze oefening gebruik je de chocolate-gegevensset om een regressieboom te fitten. Dit lijkt sterk op wat je al in hoofdstuk 1 deed met de diabetes-gegevensset.

De trainingsdata chocolate_train zijn beschikbaar in je werkruimte.

Deze oefening maakt deel uit van de cursus

Machine Learning met boomgebaseerde modellen in R

Cursus bekijken

Oefeninstructies

  • Bouw model_spec, een specificatie voor een regressieboom.
  • Gebruik op basis van de data frame chocolate_train model_spec om een regressieboom te trainen die final_grade voorspelt met alleen de numerieke predictors in de data.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

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
Code bewerken en uitvoeren