如何检查树是分类树还是回归树

Question

我想做一个修剪我的树木的通用方法。 我用来制作树的软件包是标准的c-ran“树”软件包。

该方法取决于我得到的树的类型。 我正在用像这样的tree方法来构建我的树

classification_tree_gini<- tree(Class~., train_df[,c(-1,-28)], split ="gini")
regression_tree_gini <- tree(UPDRS ~., train_df[,c(-1,-29)], split = "gini")

如何在代码方面检索树的类型，以便实现我的方法？

Answer 1

返回的树已存储用于训练数据预测的值。 您应该能够问这些预测的类别是什么。

class(regression_tree_gini$y)
[1] "numeric"
class(classification_tree_gini$y)
[1] "factor"