Подтвердить что ты не робот

Указание формулы в R с glm без явного объявления каждого ковариата

Я хотел бы заставить определенные переменные в регрессии glm без полного указания каждого из них. Мой реальный набор данных имеет ~ 200 переменных. До сих пор я не смог найти образцы этого в моем онлайн-поиске.

Например (всего 3 переменные):

n=200
set.seed(39) 
samp = data.frame(W1 = runif(n, min = 0, max = 1), W2=runif(n, min = 0, max = 5)) 
samp = transform(samp, # add A
A = rbinom(n, 1, 1/(1+exp(-(W1^2-4*W1+1))))) 
samp = transform(samp, # add Y
Y = rbinom(n, 1,1/(1+exp(-(A-sin(W1^2)+sin(W2^2)*A+10*log(W1)*A+15*log(W2)-1+rnorm(1,mean=0,sd=.25))))))

Если я хочу включить все основные термины, это имеет простой ярлык:

glm(Y~., family=binomial, data=samp)

Но скажу, что я хочу включить все основные термины (W1, W2 и A) плюс W2 ^ 2:

glm(Y~A+W1+W2+I(W2^2), family=binomial, data=samp)

Есть ли ярлык для этого?

[редактирование себя перед публикацией:] Это работает! glm(formula = Y ~ . + I(W2^2), family = binomial, data = samp)

Хорошо, так что насчет этого!

Я хочу опустить одну основную переменную терминов и включить только два основных члена (A, W2) и W2 ^ 2 и W2 ^ 2: A:

glm(Y~A+W2+A*I(W2^2), family=binomial, data=samp)

Очевидно, что с помощью нескольких переменных не требуется ярлык, но я работаю с данными высокого уровня. В текущем наборе данных есть только "200", но некоторые другие имеют тысячи и тысячи.

4b9b3361

Ответ 1

Ваше творческое использование . для создания формулы, содержащей все или почти все переменные, является хорошим и чистым подходом. Другим вариантом, который иногда бывает полезным, является построение формулы программно как строка, а затем преобразование ее в формулу с помощью as.formula:

vars <- paste("Var",1:10,sep="")
fla <- paste("y ~", paste(vars, collapse="+"))
as.formula(fla)

Конечно, вы можете сделать объект fla более сложным.

Ответ 2

Анико ответила на ваш вопрос. Чтобы увеличить бит:

Вы также можете исключить переменные, используя -:

glm(Y~.-W1+A*I(W2^2), family=binomial, data=samp)

Для больших групп переменных я часто делаю фрейм для группировки переменных, что позволяет вам сделать что-то вроде:

vars <- data.frame(
    names = names(samp),
    main = c(T,F,T,F),
    quadratic =c(F,T,T,F),
    main2=c(T,T,F,F),
    stringsAsFactors=F
)


regform <- paste(
    "Y ~",
    paste(
      paste(vars[vars$main,1],collapse="+"),
      paste(vars[1,1],paste("*I(",vars[vars$quadratic,1],"^2)"),collapse="+"),
      sep="+"
    )
)
> regform
[1] "Y ~ W1+A+W1 *I( W2 ^2)+W1 *I( A ^2)"

> glm(as.formula(regform),data=samp,family=binomial)

Используя все типы условий (по имени, по структуре и т.д.) для заполнения фрейма данных, я могу быстро выбрать группы переменных в больших наборах данных.