【问题标题】:Obtain variables used by Random Forest in R [duplicate]获取R中随机森林使用的变量[重复]
【发布时间】:2019-05-22 02:03:18
【问题描述】:

目前,我在 R 中创建了一个随机森林模型,名为:

my_rforest

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

my_var <- my_rforest$importance

这给了我输出:

                      MeanDecreaseGini
temperature           6.51
wind                  4.67
heat                  0.3
.
.
.

有什么方法可以获得变量列,即输出的第一列?

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

library(randomForest)
library(caret)       

【问题讨论】:

标签: r


【解决方案1】:

我们可以使用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)

【讨论】:

  • 您能解释一下第二行中的 [,5,drop=FALSE] 是什么意思吗?
  • @Maxxx 只是为了与您发布的示例保持一致。如果您查看iris.rf$importance,还有其他列。我提取了第 5 列的“MeanDecreaseGini”,默认情况下,?Extract 使用drop = TRUE 在有单列/行时删除维度。因此,为了防止使用 drop = FALSE
猜你喜欢
  • 2020-06-06
  • 2019-02-25
  • 1970-01-01
  • 2015-04-24
  • 2017-08-29
  • 2016-06-24
  • 2017-08-29
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多