開始使用免費開始

擬合隨機森林

就像你在影片中看到的,隨機森林模型比線性模型彈性高得多,既能擬合複雜的非線性效果,也能自動捕捉變數之間的交互作用。它們在真實世界資料上通常表現很好。我們就用葡萄酒品質資料集來試試看吧。目標是根據一批葡萄酒的機器量測化學與物理性質,來預測人工評分的品質。

擬合隨機森林模型的流程,和你在上一章做的廣義線性迴歸模型完全一樣。你只需要把 train 函式中的 method 參數改成 "ranger"ranger 套件是 R 經典 randomForest 套件的重寫版,擬合速度快很多,但結果幾乎相同。我們建議所有初學者在做隨機森林建模時使用 ranger 套件。

本練習屬於課程

使用 R 的 caret 進行 Machine Learning

檢視課程

練習說明

  • 在葡萄酒品質資料集 wine 上訓練一個名為 model 的隨機森林,使 quality 為應變數,其他所有變數為解釋變數。
  • 使用 method = "ranger"
  • 使用 tuneLength 為 1。
  • 使用 5 折交叉驗證(CV folds)。
  • 在主控台列印 model

動手互動練習

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

# Fit random forest: model
model <- train(
  ___,
  tuneLength = ___,
  data = ___, 
  method = ___,
  trControl = trainControl(
    method = "cv", 
    number = ___, 
    verboseIter = TRUE
  )
)

# Print model to console
編輯並執行程式碼