Aan de slagGa gratis aan de slag

Teken de ROC-curve

Door de modelprestatie te visualiseren met een ROC-curve bundel je de prestaties over alle mogelijke drempels in één grafiek. Je ziet de sensitiviteit en specificiteit voor elke drempel. Hoe meer een ROC-curve "omhoog en naar links" ligt, hoe beter het model is.

Je gaat de klassenkansen voorspellen voor creditcardklanten die zijn opgezegd en de resultaten als een ROC-curve plotten.

Vooraf geladen zijn een model (een beslisboom getraind op de trainingsset van de creditcardklanten) en de testgegevens, customers_test.

Deze oefening maakt deel uit van de cursus

Machine Learning met boomgebaseerde modellen in R

Cursus bekijken

Oefeninstructies

  • Gebruik model om klassenkansen op de testset te voorspellen.
  • Voeg de resultaten toe aan de testset met bind_cols() en sla het op als predictions.
  • Bereken de ROC-curve van het resultaat.
  • Plot de ROC-curve met autoplot().

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Predict probabilities on test set
predictions <- predict(___, 
                       ___, 
                       type = "___") %>% 
  # Add test set
  ___(customers_test)

# Calculate the ROC curve for all thresholds
roc <- ___(___,
           estimate = ___, 
           truth = ___)

# Plot the ROC curve
___
Code bewerken en uitvoeren