r - 从决策树中提取使用的列名/特征
问题描述
有人可以这么好心并指出一种提取拟合树中使用的列/特征的方法,使用如下代码:
library(dplyr)
library(caret)
library(rpart)
df <- data.frame(
x1 = rnorm(100, 3, 1)
, x2 = rnorm(100, 5, 2)
, y = rnorm(100, 3, 1)
)
ivs <- df %>%
dplyr::select(
x1
, x2
)
fitted <- caret::train(
x = ivs
, y = df$y
)
model <-fitted$finalModel
model
这个:
model$terms
为您提供所有术语(以 x 表示)-因此不一定是曾经使用过的。
PS:
沿着这些思路的东西可能是前进的方向:
model$frame %>%
select(
var
) %>%
filter(
var != "<leaf>"
)
解决方案
随机森林模型是决策树的集合,您可以使用该函数varUsed
获取树使用的变量。如果您使用插入符号,则需要设置keep.forest=TRUE
:
library(caret)
set.seed(111)
X = matrix(runif(2000),ncol=20)
colnames(X) = paste0("col",1:20)
y = rnorm(100)
fitted <- train(x = X,y = y,trControl = trainControl(method="cv"),
keep.forest=TRUE,tuneGrid = data.frame(mtry=3:5))
我们看看使用的变量,下表告诉你每个变量(行),它在每棵树(列)中使用了多少次:
model <-fitted$finalModel
varUsed(model,by.tree=TRUE)
dim(varUsed(model,by.tree=TRUE))
[1] 20 500
head(varUsed(model,by.tree=TRUE))
[,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] [,11] [,12] [,13] [,14]
[1,] 1 3 1 4 1 2 2 2 2 2 1 0 3 2
[2,] 2 4 2 2 3 3 1 4 1 3 1 4 0 4
[3,] 3 1 5 1 1 0 2 2 0 1 2 1 2 1
[4,] 4 0 2 3 1 0 2 1 4 2 1 2 1 1
[5,] 0 0 1 0 1 4 3 2 2 2 3 4 0 0
[6,] 1 2 1 3 2 2 2 1 1 1 1 1 2 3
您可以使用以下方法详细检查:
head(getTree(model,1,labelVar=TRUE))
left daughter right daughter split var split point status prediction
1 2 3 col13 0.9651637 -3 0.04972823
2 4 5 col13 0.5825061 -3 -0.05466984
3 0 0 <NA> 0.0000000 -1 2.55528193
4 6 7 col6 0.8524606 -3 -0.30132135
5 8 9 col16 0.8011010 -3 0.51518018
6 10 11 col3 0.3629737 -3 -0.45289996
如果我们将该split var
列制成表格,我们会得到与上一个表相同的结果。
推荐阅读
- python - 将 SAS 数据文件读取到 Spark Dataframe 并加载到 S3 时出现问题
- javascript - 烧瓶登录重定向生成 302 但浏览器不会像其他功能一样更新
- linux - 我可以创建 udev 规则来跟踪 edac 错误吗?
- sql - PL/SQL Case When - 将输入映射到其他列值
- python-3.x - 使用枕头生成可搜索的 PDF
- linux - Jenkins 不能使用 aws-cli 密钥。无法找到 AWS CLI 的凭证
- google-apps-script - 如何使用 google sheet api v4 获取 protectedRangeId?
- javascript - 具有相同值的多个选择执行此操作
- google-app-engine - 是否可以与 Google App Engine 并行部署服务?
- node.js - Node.JS Express - POST 请求不起作用(返回 404) - 在端口 8080 的子目录中运行