r - 树中使用的变量

Question

我如何知道构建的树中实际使用了哪些变量？

model = tree(status~., set.train)

如果我写的话我可以看到变量:

summary(model)

tree(formula = status ~ ., data = set.train)
Variables actually used in tree construction:
[1] "spread1"      "MDVP.Fhi.Hz." "DFA"          "D2"           "RPDE"                "MDVP.Shimmer" "Shimmer.APQ5"
Number of terminal nodes:  8 
Residual mean deviance:  0.04225 = 5.831 / 138 
Distribution of residuals:
   Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
-0.9167  0.0000  0.0000  0.0000  0.0000  0.6667 

但是我怎样才能得到一个向量，其中实际使用了哪些变量的索引？

Answer 1

您可以使用str()函数查看对象的结构。在那里查看时，您应该会看到几个不同的位置来提取用于制作树模型的变量，以下是一个示例:

> library(tree)
> 
> fit <- tree(Species ~., data=iris)
> attr(fit$terms,"term.labels")
[1] "Sepal.Length" "Sepal.Width"  "Petal.Length" "Petal.Width"

编辑:既然您特别要求提供索引，您只需match()那些返回数据集中的变量名称(尽管它们可能总是按顺序排列 - 我没有使用过tree 包之前所以我不能说)。

> match(attr(fit$terms,"term.labels"),names(iris))
[1] 1 2 3 4
> names(iris)[match(attr(fit$terms,"term.labels"),names(iris))]
[1] "Sepal.Length" "Sepal.Width"  "Petal.Length" "Petal.Width" 

编辑2:

你说得对!试试这个:

> summary(fit)$used
[1] Petal.Length Petal.Width  Sepal.Length
Levels: <leaf> Sepal.Length Sepal.Width Petal.Length Petal.Width