是否有一种方法可以更改CART决策树拆分标签,而无需更改数据框中的原始变量名称和/或值标签?我试过使用我从“ prp”文档中复制出来的split.function(它似乎也可以作为rpart.plot()函数中“ split.fun =“的选项)工作,但是我无法获取它可以做多个替换。回去重新为此命名所有变量有点麻烦。那是我唯一的选择吗?
编辑(希望更为具体):有没有一种方法可以使该函数替换多个标签-说如果我有2个变量Nstage和tstage,并且我希望它们在树中的标签为“ n-stage”,并且“ t-stage”?
split.function <- function(x, labs, digits, varlen, faclen)
{
# replace commas with spaces (needed for strwrap)
labs <- gsub("Nstage", "n-stage", labs)
for(i in 1:length(labs)) {
# split labs[i] into multiple lines
labs[i] <- paste(strwrap(labs[i], width = 15), collapse = "\n")
}
labs }