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

Question

我想制作修剪树木的通用方法.我用来制作树的包是标准的c-ran"树"包.

该方法取决于我得到的树的类型.我正在使用树方法构建我的树

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"