Krzywa ROC
Czas stworzyć krzywą ROC dla naszego klasyfikatora lasu losowego. Pierwszym krokiem jest obliczenie prawdopodobieństw przewidywanych przez klasyfikator dla każdej etykiety – służy do tego metoda .predict_proba(). Następnie możesz użyć funkcji roc_curve z sklearn.metrics, aby wyliczyć współczynnik fałszywych alarmów i czułość modelu, które potem zwizualizujesz za pomocą matplotlib.
Klasyfikator RandomForestClassifier wytrenowany na 70% danych jest już dopasowany do zbioru i dostępny w twoim środowisku jako clf.
To ćwiczenie jest częścią kursu
Marketing Analytics: Przewidywanie rezygnacji klientów w Pythonie
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Generate the probabilities
y_pred_prob = ____.____(____)[:, 1]