如何将异方差性包含在lme混合模型的预测函数中

时间:2015-12-16 22:02:59

标签: r ggplot2 predict mixed-models nlme

合作所有人,

我绘制了一个来自" nlme"的混合模型。用" ggplot2"我想在模型中考虑我的类别变量的不同方差(weights = varIdent(form = ~1 | category))。当我检查绘图时,我意识到绘图中的模型预测没有包括异方差性,尽管模型确实如此。如何调整预测函数,包括异方差性?

创建一些数据:

no <- 1:20
type <- factor("A","B")
set.seed(1)
baseline <- rnorm(20, 30, 10)
set.seed(2)
event <- rnorm(20, 60, 30)
set.seed(3)
postevent1 <- rnorm(20, 40, 20)
set.seed(4)
postevent2 <- rnorm(20, 30, 10)
set.seed(5)
postevent3 <- rnorm(20, 20, 5)
data <- data.frame(no, baseline, event, postevent1, postevent2, postevent3)
data$type[data$no %% 2 == 0] <- "A"
data$type[data$no %% 2 != 0] <- "B"
data$event[data$type == "A"] <- data$event[data$type == "A"] -2
data$no <- factor(data$no)
data$type <- factor(data$type)

将数据转换为长格式

library(dplyr)

long <- data %>% gather(key = category, value = measure, -no, -type)

创建模型

library(nlme)
model <- lme(measure ~ category*type, random = ~ 1|no, data= long,   
weights=varIdent(form = ~1 | category), method = "REML")
new <- long %>% select(-measure)
long$pred <- predict(model, newdata = new)

绘制它

library(ggplot2)
ggplot(long, aes(x=category, y =  measure, colour = type,    
group=interaction(type,category))) +
geom_point() +
facet_grid(~ type) +
geom_line(aes(y=pred), size=0.8, colour = "black") +
theme(axis.text.x = element_text(angle=30, hjust=1, vjust=1))

enter image description here

1 个答案:

答案 0 :(得分:1)

这是我使用&#34; AICcmodavg&#34;的解决方案。包。如果有人找到更好的解决方案,请告诉我。

预测值

library(AICcmodavg)
pred <- predictSE.lme(model, newdata = new, print.matrix = TRUE)
pred <- data.frame(pred)
newlong <- data.frame(long, pred)

使用预测的95%置信区间绘制平均值

ggplot(newlong, aes(x= category, y= measure)) + 
geom_point(aes(x = category, y = fit)) + 
facet_grid(~ type) + 
geom_point(aes(colour = type), alpha = 0.5, size = 3)  + 
geom_errorbar(aes(ymax=fit + se.fit*1.96, ymin=fit - se.fit*1.96), 
position=position_dodge(0.9), width=0.25) +
theme(axis.text.x = element_text(angle=30, hjust=1, vjust=1))

enter image description here