訓練你的第一棵迴歸樹
在這個練習中,你將訓練一棵迴歸樹,使用所有 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
____.____(____, ____)