Выполните выбор модели с шагом функцией и запись результата в 0/1 вектор (R)

Question

Я пытаюсь написать алгоритм, который выполняет следующее: в R

1. На набор данных dat Используйте step функции для выполнения glm Выбор модели j ковариатов из набора переменных J-кандидата
2. Взять окончательный call из j вариантов и сравнить с полным вектором J. Записать результат в вектор 1xJ, где 1 указывает, что переменная находится в окончательном call и 0 elsewise.

Пример:

В следующем примере трех переменных (x,y,z) являются кандидатами для прогнозирования переменной dep. Step используется для выбора переменных. Моя цель состоит в том, чтобы наконец иметь вектор, указывающий, какая из входных переменных заканчивается в окончательной модели, так что здесь c(1,0,1).

n=1000 
x <- rnorm(n,0,1) 
y <- rnorm(n,0,1) 
z <- rnorm(n,0,1) 

dep <- 1 + 2 * x + 3* z + rnorm(n,0,1) 

m<-step(lm(dep~x+y+z),direction="backward") 

У меня есть трудности, извлекая имена переменных из конечного m$call и создания вектора.

Answer 1

Я думаю, что это делает это:

n=1000 

x <- rnorm(n,0,1) 
y <- rnorm(n,0,1) 
z <- rnorm(n,0,1) 

dep <- 1 + 2*x + 3*z + rnorm(n,0,1) 

m<-step(lm(dep~x+y+z),direction="backward") 

matt <- attributes(m$terms) 
matt$term.labels 
#[1] "x" "z" 

v <- c("x","y","z") 
as.integer(v %in% matt$term.labels) 
#[1] 1 0 1