如何使用文本绘制因子变量rpart的水平而不是标签/指数?

时间:2015-03-13 22:48:10

标签: r plot tree rpart

在因子变量上绘制树分割会产生一个图,其中节点标有因子的索引而不是级别的文本。如何绘制树,其中标签是因子变量的实际级别?

例如,这是足球运动员的样本数据。这个想法是国家(通过目标变量)解释了球员的收入。国家用作因子变量。

goals <- rpois(n = 5,lambda = 2)
income <- rnorm(100, goals*100000, sd = 1+goals*1000)
country <- rep(c("England","USA", "Nigeria", "Argentia", "Belgium"),
               times = 20)
player_df <- data.frame(income = income, country = country)

t1 <- rpart(income ~ ., 
      data = player_df, 
      cp = 0.01)

plot(t1); text(t1,)

以下是产生的情节:

enter image description here

1 个答案:

答案 0 :(得分:3)

试试text(t1, pretty=FALSE)。这样可以防止因子标签缩短。