開始使用免費開始

評估最佳的隨機森林

在本課程的最後一個練習中,你將評估 grid_rf 最佳模型在測試集上的 RMSE。

資料集已為你載入並完成前處理,且已分割為 80% 訓練與 20% 測試。環境中提供了 X_testy_test,以及從 sklearn.metrics 匯入並以別名 MSE 提供的函式 mean_squared_error。另外,我們也已載入你在前一題建立並訓練好的 GridSearchCV 物件 grid_rf。請注意,grid_rf 的訓練方式如下:

grid_rf.fit(X_train, y_train)

本練習屬於課程

Machine Learning with Tree-Based Models in Python

檢視課程

練習說明

  • sklearn.metrics 匯入 mean_squared_error,並命名為 MSE

  • grid_rf 取出最佳估計器,指定給 best_model

  • 使用 best_model 對測試集進行預測,將結果指定為 y_pred

  • 計算 best_model 在測試集的 RMSE。

動手互動練習

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

# Import mean_squared_error from sklearn.metrics as MSE 
____

# Extract the best estimator
best_model = ____

# Predict test set labels
y_pred = ____

# Compute rmse_test
rmse_test = ____

# Print rmse_test
print('Test RMSE of best model: {:.3f}'.format(rmse_test)) 
編輯並執行程式碼