树如何知道要构建分类树还是回归树？

Question

我正在学习分类/回归构建树，并且想了解树函数如何知道是构建分类树还是回归树。

以下内容将构建分类树：

library(tree)
library(ISLR)
library(dplyr)

Carseats <- Carseats %>% mutate(High = factor(ifelse(Sales <= 8, "No", "Yes")))
tree.carseats <- tree(High~ . -Sales, Carseats)

这会创建一个回归树：

library(MASS)
set.seed(1)
tree.boston=tree(medv~ .,Boston)

对我来说，两个对tree的调用看起来都一样。是否根据要预测的目标类型确定这一点？

Answer 1

尽管我同意duckmayr，但我发现https://cran.r-project.org/web/packages/tree/tree.pdf的文档几乎隐藏了 tree函数中的formula参数描述为：

"A formula expression. The left-hand-side (response) should be either a numerical vector when a regression tree will be fitted or a factor, when a classification tree is produced."

因此，您的假设是正确的，如果目标是一个因素，则拟合分类树，否则拟合回归树。