LoslegenKostenlos starten

ROC-Kurve

Erstellen wir jetzt eine ROC-Kurve für unseren Random-Forest-Klassifikator. Der erste Schritt ist, mit der Methode .predict_proba() die vorhergesagten Wahrscheinlichkeiten zu berechnen, die der Klassifikator für jedes Label ausgibt. Anschließend kannst du mit der Funktion roc_curve aus sklearn.metrics die False-Positive-Rate und True-Positive-Rate berechnen und diese dann mit matplotlib plotten.

Ein RandomForestClassifier mit einer Trainingssatzgröße von 70 % wurde auf die Daten gefittet und steht dir in deinem Workspace als clf zur Verfügung.

Diese Übung ist Teil des Kurses

<Kurs>Marketing Analytics: Kundenabwanderung in Python vorhersagen</Kurs>
Kurs ansehen

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# Generate the probabilities
y_pred_prob = ____.____(____)[:, 1]
Code bearbeiten und ausführen