開始使用免費開始

搜尋最佳決策樹

在這個練習中,你將使用 5-fold 交叉驗證進行網格搜尋,找出 dt 的最佳超參數。請注意,網格搜尋會全面嘗試所有組合,因此訓練模型可能需要較長時間。這裡你只需建立 GridSearchCV 物件,先不要把它配適到訓練集。正如影片所示,你可以像使用任何 scikit-learn 估計器一樣,透過 .fit() 方法來訓練這個物件:

grid_object.fit(X_train, y_train)

未調校的分類樹 dt,以及你在前一個練習中定義的字典 params_dt,都已經在你的工作環境中可用。

本練習屬於課程

Machine Learning with Tree-Based Models in Python

檢視課程

練習說明

  • sklearn.model_selection 匯入 GridSearchCV

  • 以 5-fold CV 建立一個 GridSearchCV 物件,並設定參數:

    • estimator 設為 dtparam_grid 設為 params_dt,以及

    • scoring 設為 'roc_auc'

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Import GridSearchCV
____

# Instantiate grid_dt
grid_dt = ____(estimator=____,
                       param_grid=____,
                       scoring=____,
                       cv=____,
                       n_jobs=-1)
編輯並執行程式碼