如何檢查樹是分類樹還是回歸樹

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"