如何编辑多个rpart.plot()拆分标签?

时间:2018-10-19 19:45:11

标签: cart decision-tree rpart

是否有一种方法可以更改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 }

0 个答案:

没有答案