IniziaInizia gratis

ROC e AUC in-sample

Quanto bene gli alberi bagged catturano la struttura del tuo insieme di addestramento? Sono migliori degli alberi di decisione? Fanno overfitting? Usare ROC e AUC è un ottimo modo per valutarlo.

In questo esercizio, genererai queste previsioni in-sample e calcolerai ROC e AUC. Attenzione, ci saranno sorprese!

Sono già caricati il risultato dell'esercizio precedente, model_bagged, e i dati di addestramento, customers_train.

Questo esercizio fa parte del corso

Machine Learning con modelli ad albero in R

Visualizza il corso

Istruzioni dell'esercizio

  • Usa model_bagged per generare previsioni di probabilità con il tuo insieme di addestramento e aggiungile al tibble di training, salvando il risultato come predictions.
  • Genera la curva ROC del tibble predictions e traccia il risultato.
  • Calcola l'AUC del tibble predictions.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Predict on training set and add to training set
predictions <- ___(___,
                   new_data = ___, 
                   type = "___") %>% 
    bind_cols(___)

# Create and plot the ROC curve
roc_curve(___,
          estimate = ___,
          truth = ___) %>% autoplot()

# Calculate the AUC
___(predictions,
    estimate = ___, 
    truth = ___)
Modifica ed esegui il codice