Como plotar um plano de regressão com uma interação no rgl
Eu quero plotar a superfície de regressão de um modelo com um termo de interação usandorgl
sistema interativo de plotagem. É fácil traçar um plano de regressão para um modelo sem um termo de interação usando:
plot3d(x=x1, y=x2, z=y1, type="s", col="yellow", size=1)
planes3d(a=coef(mod1)[2], b=coef(mod1)[3], c=-1, d=coef(mod1)[1], alpha=.5)
No entanto, quando o avião torce, isso parece ser mais difícil. Seguindo essa pergunta:Equivalente 3D da função de curva em r, Eu estou tentando:
f2 <- function(x, y) as.vector(coef(mod2)%*%c(1, x, y, x*y))
curve_3d <- function(f2, x_range=c(0, 40), y_range=c(0, 40)){
if (!require(rgl) ) {stop("load rgl")}
xvec <- seq(x_range[1], x_range[2], by=1)
yvec <- seq(y_range[1], y_range[2], by=1)
fz <- outer(xvec, yvec, FUN=f2)
persp3d(xvec, yvec, fz, alpha=.5)
}
open3d()
plot3d(x=x1, y=x2, z=y2, type="s", col="yellow", size=1)
curve_3d(f2)
Mas não está funcionando. (Eu tentei algumas outras coisas também, mas estou mantendo isso curto.) Meu principal problema até agora parece ser comf2
; no entanto, eu também quero que isso pareçaplanes3d
e eu não tenho certeza se isso vai me dar um wireframe.
Aqui está um exemplo:
set.seed(897)
x1 = rep(c(0, 10, 20, 30, 40), times=25)
x2 = rep(c(0, 10, 20, 30, 40), each=25)
y2 = 37 + 0.7*x1 + 1.2*x2 - 0.05*x1*x2 + rnorm(125, mean=0, sd=5)
mod2 = lm(y2~x1*x2)
open3d()
plot3d(x=x1, y=x2, z=y2, type="s", col="yellow", size=1)
curve_3d(f2)