Szukaj optymalnego drzewa
W tym ćwiczeniu przeprowadzisz przeszukiwanie siatki (grid search) z użyciem 5-krotnej kroswalidacji, aby znaleźć optymalne hiperparametry modelu dt. Pamiętaj, że przeszukiwanie siatki jest procesem wyczerpującym i trenowanie modelu może zająć sporo czasu. W tym zadaniu tylko utworzysz obiekt GridSearchCV, bez dopasowywania go do zbioru treningowego. Jak omówiono w filmie, taki obiekt możesz wytrenować tak samo jak każdy estymator scikit-learn – za pomocą metody .fit():
grid_object.fit(X_train, y_train)
W twoim środowisku dostępne są: niedostrojone drzewo klasyfikacyjne dt oraz słownik params_dt zdefiniowany w poprzednim ćwiczeniu.
To ćwiczenie jest częścią kursu
Uczenie maszynowe z modelami drzewiastymi w Pythonie
Instrukcje do ćwiczenia
Zaimportuj
GridSearchCVzsklearn.model_selection.Utwórz obiekt
GridSearchCVz 5-krotną kroswalidacją, ustawiając następujące parametry:estimatornadt,param_gridnaparams_dtorazscoringna'roc_auc'.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Import GridSearchCV
____
# Instantiate grid_dt
grid_dt = ____(estimator=____,
param_grid=____,
scoring=____,
cv=____,
n_jobs=-1)