时间戳

首页 > 解决方案 > 获取 R 中随机森林使用的变量

r - 获取 R 中随机森林使用的变量

问题描述

目前，我在 R 中创建了一个随机森林模型，称为：

my_rforest

我正在尝试访问我的数据集的随机森林使用的变量，但到目前为止，我做到了：

my_var <- my_rforest$importance

这给了我输出：

                      MeanDecreaseGini
temperature           6.51
wind                  4.67
heat                  0.3
.
.
.

有什么办法可以获取变量列，即输出的第一列？

我目前正在为我的随机森林使用以下软件包。

library(randomForest)
library(caret)

标签： r

解决方案

我们可以row.names用来获取变量

library(randomForest)
my_var <- iris.rf$importance[,5, drop = FALSE]
row.names(my_var)
#[1] "Sepal.Length" "Sepal.Width"  "Petal.Length" "Petal.Width"

注意：使用可重现的示例

数据

set.seed(71)
iris.rf <- randomForest(Species ~ ., data=iris, importance=TRUE,
                     proximity=TRUE)

推荐阅读