開始使用免費開始

設定參數並訓練模型

預測任務通常分為兩類:迴歸或分類。在這個糖果資料集中,輸出變數是「連續」型,描述在一系列 1 對 1 對決中,某個糖果被選擇勝過另一個糖果的頻率。為了預測這個數值(勝率),你將使用「迴歸」模型。

在本練習中,你會使用隨機森林迴歸模型 rfr,並指定幾個參數。

本練習屬於課程

Python 的模型驗證

檢視課程

練習說明

  • rfr 中加入參數,將樹的數量設為 100,且每棵樹的最大深度設為 6。
  • 為了讓結果可重現,將隨機種子設為 1111
  • 使用 .fit() 方法訓練隨機森林迴歸模型,將 X_train 作為輸入資料,y_train 作為目標。

動手互動練習

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

# Set the number of trees
rfr.____ = ____

# Add a maximum depth
rfr.____ = ____

# Set the random state
rfr.____ = ____

# Fit the model
rfr.____(____, ____)
編輯並執行程式碼