Disegna la curva ROC
Visualizzare le prestazioni del modello con una curva ROC ti permette di riunire in un unico grafico le performance su tutte le possibili soglie. Mostra sensibilità e specificità per ogni soglia. Più una curva ROC è spostata "in alto a sinistra", migliore è il modello.
Prevederai le probabilità di classe dei clienti della carta di credito che hanno fatto churn e rappresenterai i risultati come curva ROC.
Sono già caricati un model, ovvero un albero decisionale addestrato sul training set dei clienti con carta di credito, e i dati di test, customers_test.
Questo esercizio fa parte del corso
Machine Learning con modelli ad albero in R
Istruzioni dell'esercizio
- Usa
modelper prevedere le probabilità di classe sul set di test. - Aggiungi i risultati al set di test usando
bind_cols()e salva il risultato comepredictions. - Calcola la curva ROC del risultato.
- Traccia la curva ROC usando
autoplot().
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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
___