R插入符号随机森林ggplot修改图例

问题描述

如何修改通过将 ggplot 应用于使用 Ranger 算法构建的插入符号对象生成的默认图例？例如，假设我希望图例标题为“拆分算法”，而不是默认的“拆分规则”。

library(caret)

trainIndex <- createDataPartition(iris$Species,p = 0.80,list = FALSE)

train <- iris[ trainIndex,]

test  <- iris[-trainIndex,]

rfs <- train(Species ~ .,data = train,trControl = trainControl(method = "cv",number = 10),method = "ranger",tuneGrid = expand.grid(mtry = seq(1,3,1),splitrule = c("gini","extratrees"),min.node.size = 1))

ggplot(rfs) + 
  xlab("Number of parameters") +
  ylab("Accuracy") +
  theme(legend.title = "Splitting algo")

解决方法

您使用 name 和 scale_color_discrete 设置了 scale_shape_discrete 的图例。

library(ggplot2)

ggplot(rfs) + 
  xlab("Number of parameters") +
  ylab("Accuracy") +
  scale_color_discrete(name = "Splitting algo")  +
  scale_shape_discrete(name = "Splitting algo")