Kom igångKom igång gratis

Gör förutsägelser

Att göra förutsägelser med data är ett av de grundläggande målen inom maskininlärning. Nu när du vet hur man delar upp data och anpassar en modell är det dags att göra förutsägelser om okända urval med hjälp av dina modeller.

Du ska göra förutsägelser om ditt testset med en modell som skapas genom att anpassa träningsdatan till en trädspecifikation.

I din workspace finns de datamängder du skapade tidigare (diabetes_train och diabetes_test) samt en beslutsträdsspecifikation tree_spec, som skapades med följande kod:

tree_spec <- decision_tree() %>%
  set_engine("rpart") %>%
  set_mode("classification")

Den här övningen är en del av kursen

Maskininlärning med trädbaserade modeller i R

Visa kurs

Övningsinstruktioner

  • Anpassa din specifikation till träningsdatan med outcome som målvariabel och alla prediktorer för att skapa model.
  • Använd din modell för att förutsäga diabetesutfallet för varje observation i testsetet och tilldela resultatet till predictions.
  • Lägg till det sanna utfallet från testsetet i predictions som en kolumn med namnet true_class och spara resultatet som predictions_combined.
  • Använd funktionen head() för att skriva ut de första raderna i resultatet.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Train your model
model <- tree_spec %>% 
  ___

# Generate predictions
predictions <- ___(model,
                   ___)

# Add the true outcomes
predictions_combined <- predictions %>% 
  ___(true_class = ___)

# Print the first lines of the result
___
Redigera och kör kod