Caret的重要性

时间:2018-06-10 16:11:38

标签: r logistic-regression r-caret

我在R中使用Caret包来训练二元分类问题的逻辑回归模型。我已经能够得到结果,准确性等,但我也想要变量的重要性(按重要性递减顺序)。我使用了varImp()函数。但根据文档,重要性取决于班级:

“对于大多数分类模型,每个预测变量对每个类都有一个单独的变量重要性(例外是分类树,袋装树和增强树)。”

如何获得每个班级的变量重要性?

谢谢

1 个答案:

答案 0 :(得分:0)

对于第一部分,您是否尝试过:

round(importance(myModel$finalModel), 2)

要按降序排列:

imp <- round(importance(myModel$finalModel), 2)
dfimp <- data.frame(feature = rownames(imp), MeanDecreaseGini = as.numeric(imp))
dfimp[order(dfimp$MeanDecreaseGini, decreasing = T),]