Začněte nyníZačněte zdarma

Hledání optimálního stromu

V tomto cvičení použiješ grid search s 5-násobnou křížovou validací k nalezení optimálních hyperparametrů modelu dt. Měj na paměti, že grid search je vyčerpávající proces a trénování modelu může trvat delší dobu. Tady se zaměříš pouze na vytvoření objektu GridSearchCV bez jeho trénování na trénovací sadě. Jak bylo ukázáno ve videu, takový objekt lze trénovat stejně jako jakýkoli estimátor ze scikit-learn pomocí metody .fit():

grid_object.fit(X_train, y_train)

V tvém pracovním prostoru je k dispozici neoptimalizovaný klasifikační strom dt a slovník params_dt, který jsi definoval/a v předchozím cvičení.

Toto cvičení je součástí kurzu

Machine Learning with Tree-Based Models in Python

Zobrazit kurz

Pokyny k cvičení

  • Importuj GridSearchCV z sklearn.model_selection.

  • Vytvoř objekt GridSearchCV s 5-násobnou křížovou validací nastavením těchto parametrů:

    • estimator na dt, param_grid na params_dt a

    • scoring na 'roc_auc'.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Import GridSearchCV
____

# Instantiate grid_dt
grid_dt = ____(estimator=____,
                       param_grid=____,
                       scoring=____,
                       cv=____,
                       n_jobs=-1)
Upravit a spustit kód