从数据框创建虚拟变量

时间:2018-10-18 17:39:29

标签: r bayesian

我需要做一个贝叶斯线性回归(使用重新考虑软件包),它要求我的分类变量是数字(我认为),所以我试图创建虚拟变量,以便用(0,1, 2,3 ...)。我在下面模拟了一些数据,向您展示了我的数据。请记住,我的实际数据包含更多在此处表示的分类变量,因此使用一堆ifelse函数并不理想:

#simulated data
X <- data.frame(
    Longest.axis = rnorm(50,10,5),
    Time..hrs. = round(runif(50,0,4)),
    Strain = sapply(1:50,function(x) {ifelse(x<12,"Msmeg_2788","Msmeg_WT")}),
    Item.Name = sapply(1:50,
        function(x) 
            {ifelse(x<6,"Q109.jpg",
            ifelse(x<20,"Q340.jpg",
                ifelse(x<40,"Q11.jpg","Q230.jpg")
            )
        )}
    )
) 

下面我展示了创建虚拟变量的尝试:

#create dummy variables
Straindummy <- model.matrix(X$Strain)
Item.Namedummy <- model.matrix(X$Item.Name)

它返回以下错误:

Error: $ operator is invalid for atomic vectors

这是我尝试进行多元贝叶斯线性回归的尝试,以防万一您需要它来建议我应该做什么:

library(rethinking)
model <- map(
    alist(
        X$Longest.Axis ~ dnorm(mu,sigma),
        mu <- a + b1*Straindummy+b2*Item.Namedummy+b3*X$Time..hrs.,
        a ~ dnorm(10,10),
        b1 ~ dnorm(0,1),
        b2 ~ dnorm(0,1),
        b3 ~ dnorm(0,1),
        sigma ~ dunif(0,10)
    ),
    data = X )

谢谢您能提供任何帮助!

1 个答案:

答案 0 :(得分:2)

您不需要做任何model.matrix东西,因为R因子是“底层”的数字。但是您确实需要安装(并且可能已经安装了Github,但可能没有为以后的读者描述如何做)Github的依赖项和软件包:

install.packages(c("coda","mvtnorm","devtools","loo"))
library(devtools)
devtools::install_github("rmcelreath/rethinking")

然后,在更正了变量名错误之后,原始代码将使用原始列名运行:

 library(rethinking)
model <- map(
    alist(
        Longest.axis ~ dnorm(mu,sigma),
        mu <- a + b1*X$Strain+b2*X$Item.Name+b3*X$Time..hrs.,
        a ~ dnorm(10,10),
        b1 ~ dnorm(0,1),
        b2 ~ dnorm(0,1),
        b3 ~ dnorm(0,1),
        sigma ~ dunif(0,10)
    ),
    data = X )

model
#--------------------------------------
Maximum a posteriori (MAP) model fit

Formula:
Longest.axis ~ dnorm(mu, sigma)
mu <- a + b1 * X$Strain + b2 * X$Item.Name + b3 * X$Time..hrs.
a ~ dnorm(10, 10)
b1 ~ dnorm(0, 1)
b2 ~ dnorm(0, 1)
b3 ~ dnorm(0, 1)
sigma ~ dunif(0, 10)

MAP values:
          a          b1          b2          b3       sigma 
10.73889375 -0.21053314  0.01836911 -0.24736940  4.69900162 

Log-likelihood: -148.32 

此外,如果您省略了访问数据的X $方法,它可以正确运行:

...
mu <- a + b1*Strain+b2*Item.Name+b3*X$Time..hrs.,
...

正确编写的函数将通过在data参数的上下文中正确扩展整个模型矩阵来解释因子变量。因此,您不必以objecName$columnName的形式访问变量,而只需提供一个带有列名和data=参数的对象的公式。这适用于数字变量和因子变量。看一下结果:

 object <- data.frame(x=runif(10), y=factor( sample(c('a','b','c'),10,rep=TRUE)))
 model.matrix(~x+y, object)
#----------------------
(Intercept)          x yb yc
1            1 0.04942913  0  0
2            1 0.92403817  1  0
3            1 0.46485707  0  1
4            1 0.57529925  1  0
5            1 0.47036022  0  1
6            1 0.63165954  0  0
7            1 0.11975911  0  1
8            1 0.13187866  1  0
9            1 0.25253765  0  1
10           1 0.52289041  1  0
attr(,"assign")
[1] 0 1 2 2
attr(,"contrasts")
attr(,"contrasts")$y
[1] "contr.treatment