開始使用免費開始

訓練你的第一棵迴歸樹

在這個練習中,你將訓練一棵迴歸樹,使用所有 6 個可用特徵,來預測 auto-mpg 資料集 中汽車的 mpg(每加侖行駛英里數)油耗。

資料集已為你處理好,並切分為 80% 訓練集與 20% 測試集。特徵矩陣 X_train 與陣列 y_train 已經在你的工作空間中可用。

本練習屬於課程

Machine Learning with Tree-Based Models in Python

檢視課程

練習說明

  • sklearn.tree 匯入 DecisionTreeRegressor
  • 建立一個 DecisionTreeRegressor 物件 dt,其最大深度為 8,且 min_samples_leaf 設為 0.13。
  • dt 擬合到訓練集。

動手互動練習

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

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
編輯並執行程式碼