在因子变量上绘制树分割会产生一个图,其中节点标有因子的索引而不是级别的文本。如何绘制树,其中标签是因子变量的实际级别?
例如,这是足球运动员的样本数据。这个想法是国家(通过目标变量)解释了球员的收入。国家用作因子变量。
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,)
以下是产生的情节:
答案 0 :(得分:3)
试试text(t1, pretty=FALSE)
。这样可以防止因子标签缩短。