首页 > 解决方案 > 如何更改点的颜色或填充 varImpPlot 函数中的点以使其更清晰?

问题描述

我使用了随机森林并尝试使用varImpPlot()绘制重要特征

varImpPlot(random_fit, type = 1, main = "RF Feature Importance")

并且这些点对眼睛来说并不是很清楚,那么我怎样才能将其更改为垂直线,或者填充这些点?

在此处输入图像描述

标签: rmachine-learningplotrandom-forest

解决方案


varImpPlot 文档中所述,该功能还可以接受:

...要传递给的其他图形参数dotchart

例如,您可以使用以下内容填充点bg

library(randomForest)

set.seed(4543)
data(mtcars)

mtcars.rf <- randomForest(mpg ~ ., data=mtcars, ntree=1000, keep.forest=FALSE,
                          importance=TRUE)

varImpPlot(mtcars.rf, bg = "skyblue")

在此处输入图像描述

增加它们的大小cex

varImpPlot(mtcars.rf, bg = "skyblue", cex=2)

在此处输入图像描述

或更改其类型pch

varImpPlot(mtcars.rf, bg = "skyblue", cex=2, pch=22)

在此处输入图像描述

AFAIK,你不能添加垂直线(要求type="l"会导致错误);这可能是因为在底层dotchart使用的基本 R 图形函数不支持线条。


推荐阅读