В функции rpart.plot (из пакета rpart.plot, расширение для пакета rpart) есть аргумент box.col
, который управляет цветом узлов в дереве. Как установить его так, чтобы он окрашивал узлы так, чтобы узлы одного и того же ответа окрашивались одинаково?
Я пробовал несколько различных вариантов с аргументом box.col
, например, используя cols
в качестве факторов, которые игнорируют выбранные цвета. Самое близкое, что у меня есть, показано ниже
set.seed(1);x <- runif(100)
set.seed(2);y <- runif(100)
data <- matrix(c(x,y),ncol=2)
fact <- as.numeric(factor(--((x > 0.5 & y < 0.5))))
fact[x < 0.1] = 3
cols <- (c("grey80", "red", "blue"))
plot(data, col=fact)
t1 <- rpart(factor(fact) ~ data)
rpart.plot(t1, type=5, extra=2,
box.col=cols)
Я ожидаю / хочу, чтобы каждый соответствующий узел ответа был окрашен одинаково. В данном коде я ожидал бы, что узлы «1» будут серыми80, «2» - красными, а «3» - голубыми. Приведенный выше график показывает, что на самом деле происходит, что не помогает.
Как уже упоминалось в первой части, как мне настроить его так, чтобы он окрашивал узлы так, чтобы узлы одного и того же ответа окрашивались одинаково?