开始使用免费开始使用

训练回归树

如您所知,决策树是解决分类问题的有用工具。此外,它也可以用于回归问题。 结构上的不同在于,叶节点上将是数值(而不是类别)。

本练习中,您将使用巧克力数据集来拟合一棵回归树。这与您在第 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
编辑并运行代码