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

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."

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