Träna ett regressionsträd
Som du redan vet är beslutsträd ett användbart verktyg för klassificeringsproblem. Du kan dessutom använda dem för att modellera regressionsproblem. Den strukturella skillnaden är att lövnoderna innehåller numeriska värden i stället för klasser.
I den här övningen använder du chokladdatasetet för att anpassa ett regressionsträd. Det liknar mycket det du redan gjorde i kapitel 1 med datasetet diabetes.
Träningsdatan chocolate_train finns tillgänglig i din arbetsmiljö.
Den här övningen är en del av kursen
Maskininlärning med trädbaserade modeller i R
Övningsinstruktioner
- Bygg
model_spec, en specifikation för ett regressionsträd. - Använd dataramen
chocolate_trainochmodel_specför att träna ett regressionsträd som förutsägerfinal_grademed enbart de numeriska prediktorerna i datan.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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