修剪后的RPART树中所选要素的名称

Question

我想获得我修剪过的RPART树中使用的功能的名称。我可以从绘制的修剪树上读取名称，但我更喜欢名字的矢量。有没有办法做到这一点？

Answer 1

需要花费一些力气，但你可以得到这个。 既然你没有提供任何例子，我会的 用一些内置数据说明。

if Date.now() - user.alive < y then removeActiveUser(user)

获取用于拆分的变量的一种方法 是通过函数library(rpart) libary(partykit) ## for nicer plotting IRP = rpart(Species ~ ., data=iris) plot(as.party(IRP)) 。

labels

很容易忽略第一个（根）节点，但我们需要 清理其他拆分的文本。我们可以使用labels(IRP) [1] "root" "Petal.Length< 2.45" "Petal.Length>=2.45" [4] "Petal.Width< 1.75" "Petal.Width>=1.75" 和一个正则表达式，只获取变量名称。

sub

如果您愿意，可以将VPat = paste0(".*(", paste(colnames(iris), collapse="|"), ").*") sub(VPat,"\\1", labels(IRP)[-1]) [1] "Petal.Length" "Petal.Length" "Petal.Width" "Petal.Width" 应用到此处以获取每个 变量名一次。