代码之家  ›  专栏  ›  技术社区  ›  Maxxx

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

r
  •  1
  • Maxxx  · 技术社区  · 6 年前

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

    my_rforest
    

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

    my_var <- my_rforest$importance
    

    这给了我输出:

                          MeanDecreaseGini
    temperature           6.51
    wind                  4.67
    heat                  0.3
    .
    .
    .
    

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

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

    library(randomForest)
    library(caret)       
    
    1 回复  |  直到 6 年前
        1
  •  1
  •   akrun    6 年前

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