У меня тут небольшая проблема, пожалуйста, помогите мне.У меня есть эти данные
set.seed(4)
mydata <- data.frame(var = rnorm(100),
temp = rnorm(100),
subj = as.factor(rep(c(1:10),5)),
trt = rep(c("A","B"), 50))
и эта модель, которая соответствует им
lm <- lm(var ~ temp * subj, data = mydata)
Я хочу построить результаты с решеткой и подгонять линию регрессии, предсказанную с помощьюмоя модель, через них.Для этого я использую этот подход, обрисованный в общих чертах «Решетки решетки для энергопотребления» Д. Саркара
temp_rng <- range(mydata$temp, finite = TRUE)
grid <- expand.grid(temp = do.breaks(temp_rng, 30),
subj = unique(mydata$subj),
trt = unique(mydata$trt))
model <- cbind(grid, var = predict(lm, newdata = grid))
orig <- mydata[c("var","temp","subj","trt")]
combined <- make.groups(original = orig, model = model)
xyplot(var ~ temp | subj,
data = combined,
groups = which,
type = c("p", "l"),
distribute.type = TRUE
)
Пока все хорошо, но я также хочу назначить цвет заливки дляТочки данных для двух обработок trt=1
и trt=2
.
Итак, я написал этот фрагмент кода, который отлично работает, но когда дело доходит до построения линии регрессии, кажется, что тип не распознается функцией панели ...
my.fill <- c("black", "grey")
plot <- with(combined,
xyplot(var ~ temp | subj,
data = combined,
group = combined$which,
type = c("p", "l"),
distribute.type = TRUE,
panel = function(x, y, ..., subscripts){
fill <- my.fill[combined$trt[subscripts]]
panel.xyplot(x, y, pch = 21, fill = my.fill, col = "black")
},
key = list(space = "right",
text = list(c("trt1", "trt2"), cex = 0.8),
points = list(pch = c(21), fill = c("black", "grey")),
rep = FALSE)
)
)
plot
Я также пытался переместить тип и распространять тип в пределах panel.xyplot
, а также подмножество данных в нем panel.xyplot
, например,
plot <- with(combined,
xyplot(var ~ temp | subj,
data = combined,
panel = function(x, y, ..., subscripts){
fill <- my.fill[combined$trt[subscripts]]
panel.xyplot(x[combined$which=="original"], y[combined$which=="original"], pch = 21, fill = my.fill, col = "black")
panel.xyplot(x[combined$which=="model"], y[combined$which=="model"], type = "l", col = "black")
},
key = list(space = "right",
text = list(c("trt1", "trt2"), cex = 0.8),
points = list(pch = c(21), fill = c("black", "grey")),
rep = FALSE)
)
)
plot
, но также безуспешно.
Может ли кто-нибудь помочь мне построить прогнозные значения в виде линий, а не точек?