開始使用免費開始

使用 Scikit Learn 的 GridSearchCV

Scikit Learn 的 GridSearchCV 模組提供許多實用功能,能有效協助你進行網格搜尋。現在請把所學付諸實作,依指定參數建立一個 GridSearchCV 物件。

需要的設定如下:

  • 使用 Random Forest 估計器,分割準則設為 'entropy'
  • 5 折交叉驗證
  • 超參數 max_depth(2、4、8、15)與 max_features('auto' 與 'sqrt')
  • 使用 roc_auc 作為模型評分指標
  • 以 4 個核心平行處理
  • 重新擬合最佳模型,並回傳訓練分數

你可以使用 X_trainX_testy_trainy_test 資料集。

本練習屬於課程

Python 超參數調校

檢視課程

練習說明

  • 依上述說明建立一個 Random Forest 估計器。
  • 依上述說明建立一個參數網格。
  • 依上述說明,使用前兩個步驟建立的物件來建立一個 GridSearchCV 物件。

動手互動練習

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

# Create a Random Forest Classifier with specified criterion
rf_class = RandomForestClassifier(____=____)

# Create the parameter grid
param_grid = {____: ____, ____: ____} 

# Create a GridSearchCV object
grid_rf_class = GridSearchCV(
    estimator=____,
    param_grid=____,
    scoring=____,
    n_jobs=____,
    cv=____,
    refit=____, return_train_score=____)
print(grid_rf_class)
編輯並執行程式碼