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
Oefeninstructies
- Gebruik
modelom klassenkansen op de testset te voorspellen. - Voeg de resultaten toe aan de testset met
bind_cols()en sla het op alspredictions. - 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
___