Beräkna AUC
I den här övningen beräknar du AUC för dina churn-prediktionsmodeller för att hitta den bästa.
Använd funktionen auc() i paketet pROC.
Funktionen tar två argument:
- Den sanna churn-etiketten i testmängden,
test_set$Future. - Modellens prediktion:
a. För logistisk regression är det prediktionen från funktionen
predict. b. För random forest är det den andra kolumnen i prediktionen från funktionenpredict.
Objekten firstPredictions, secondPredictions, thirdPredictions och rfPredictions har redan laddats in.
Vilken modell har det högsta AUC-värdet?
Den här övningen är en del av kursen
Prediktiv analys med nätverksdata i R
Interaktiv övning med praktiskt arbete
Gör teori till handling med en av våra interaktiva övningar
Starta övningen