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

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)

对我来说，两次叫树看上去都是一样的。它是否根据要预测的目标类型来确定这一点？

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

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