MulaiMulai sekarang secara gratis

Cari pohon yang optimal

Dalam latihan ini, Anda akan melakukan grid search menggunakan cross validation 5-fold untuk menemukan hiperparameter optimal milik dt. Perlu dicatat bahwa karena grid search merupakan proses yang menyeluruh, pelatihan model dapat memakan banyak waktu. Di sini Anda hanya akan membuat objek GridSearchCV tanpa melakukan fitting ke himpunan data latih. Seperti dibahas dalam video, Anda dapat melatih objek tersebut seperti estimator scikit-learn lainnya dengan metode .fit():

grid_object.fit(X_train, y_train)

Sebuah classification tree dt yang belum dituning serta kamus params_dt yang Anda definisikan pada latihan sebelumnya tersedia di workspace Anda.

Latihan ini adalah bagian dari kursus

Machine Learning dengan Model Berbasis Pohon di Python

Lihat Kursus

Petunjuk latihan

  • Impor GridSearchCV dari sklearn.model_selection.

  • Buat instance objek GridSearchCV menggunakan CV 5-fold dengan mengatur parameter:

    • estimator ke dt, param_grid ke params_dt dan

    • scoring ke 'roc_auc'.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import GridSearchCV
____

# Instantiate grid_dt
grid_dt = ____(estimator=____,
                       param_grid=____,
                       scoring=____,
                       cv=____,
                       n_jobs=-1)
Edit dan Jalankan Kode