Oceń optymalne drzewo
W tym ćwiczeniu ocenisz wartość ROC AUC na zbiorze testowym dla optymalnego modelu grid_dt.
Aby to zrobić, najpierw wyznaczysz prawdopodobieństwo uzyskania etykiety pozytywnej dla każdej obserwacji ze zbioru testowego. Możesz skorzystać z metody predict_proba() klasyfikatora sklearn, która zwraca dwuwymiarową tablicę zawierającą prawdopodobieństwa klasy negatywnej i pozytywnej – odpowiednio w kolejnych kolumnach.
Zbiór danych jest już wczytany i przetworzony (cechy numeryczne zostały wystandaryzowane); jest podzielony na 80% danych treningowych i 20% testowych. W środowisku pracy dostępne są X_test oraz y_test. Dodatkowo wczytano wytrenowany obiekt GridSearchCV o nazwie grid_dt, który został utworzony w poprzednim ćwiczeniu. Zwróć uwagę, że grid_dt był trenowany w następujący sposób:
grid_dt.fit(X_train, y_train)
To ćwiczenie jest częścią kursu
Uczenie maszynowe z modelami drzewiastymi w Pythonie
Instrukcje do ćwiczenia
Zaimportuj
roc_auc_scorezsklearn.metrics.Wyodrębnij atrybut
.best_estimator_zgrid_dti przypisz go do zmiennejbest_model.Przewidź prawdopodobieństwa uzyskania klasy pozytywnej na zbiorze testowym i zapisz je jako
y_pred_proba.Oblicz wartość ROC AUC na zbiorze testowym (
test_roc_auc) dla modelubest_model.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Import roc_auc_score from sklearn.metrics
____
# Extract the best estimator
best_model = ____
# Predict the test set probabilities of the positive class
y_pred_proba = ____
# Compute test_roc_auc
test_roc_auc = ____
# Print test_roc_auc
print('Test set ROC AUC score: {:.3f}'.format(test_roc_auc))