開始使用免費開始

訓練回歸樹

你已經知道,決策樹很適合用在分類問題。此外,它也可以用來建構回歸模型。 結構上的差異在於,葉節點上會是數值(而不是類別)。

在這個練習中,你會使用巧克力資料集來擬合一棵回歸樹。這和你在第 1 章用 diabetes 資料集做的事情非常類似。

你的工作空間中已提供訓練資料 chocolate_train

本練習屬於課程

R 的樹狀模型機器學習

檢視課程

練習說明

  • 建立 model_spec,也就是回歸樹的規格。
  • 使用資料框 chocolate_train,用 model_spec 訓練一棵回歸樹,僅使用資料中的「數值型預測變數」來預測 final_grade

動手互動練習

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

library(tidymodels)

# Build the specification
model_spec <- decision_tree() %>%
  set_mode(___) %>%
  set_engine(___)

# Fit to the data
model_fit <- model_spec %>%
  ___(formula = ___,
      data = ___)

model_fit
編輯並執行程式碼