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

Question

我想制作修剪树木的通用方法。我用来制作树木的包装是标准的c-ran＆＃34;树＆＃34;封装

该方法取决于我得到的树的类型。 我用这样的树方法构建我的树

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"