Как нарисовать поверхность регрессии с одним непрерывным предиктором и одним факторным предиктором в r? - PullRequest
Новая
       43

Как нарисовать поверхность регрессии с одним непрерывным предиктором и одним факторным предиктором в r?

0 голосов
/ 15 сентября 2018

Попытка сделать некоторую регрессию для одного непрерывного предиктора и одного факторного предиктора.

Приведенный ниже код работает при использовании abline (но получает: Предупреждение: В аблине (модели [[i]], col = labelColors [i], lty = 2: 4): только используя первые два из трех коэффициентов регрессии)) и рисует ряд линий для моделей logit и loglog.

Как мне нарисовать эту поверхность или как-то подделать? 

f<-function() {
    set.seed(123)
    colors<-c("red","green")
    labelColors<-c("black","cyan","magenta")
    x<-runif(100,0,1)
    f<-round(runif(100,0,.75))
    noise<-rnorm(length(x),0,.5)
    y<-(1+3*x+f/2+noise)/6
    plot(x,y,xlab="",ylab="")
    title("test")
    plot(x[f==0],y[f==0],ylim=c(0,1),col=colors[1])
    points(x[f==1],y[f==1], ylim=c(0.,1.),col=colors[2])
    title("test",sub="factor")
    m<-lm(y~x+factor(f),na.action=na.omit)
    logit<-betareg(y~x+factor(f),na.action=na.omit)
    loglog<-betareg(y~x+factor(f),na.action=na.omit,link="loglog")
    models<-list(m,logit,loglog)
    for(i in 1:length(models)) {  
        if(is(models[[i]],"lm")) {
            if(F) lines(x,predict(models[[i]]),col=labelColors[i],lty=2:4)
            else abline(models[[i]],col=labelColors[i],lty=2:4)
        }
        else lines(x,predict(models[[i]]),col=labelColors[i],lty=2:4)
    }
    for(i in 1:length(models))
        print(models[[i]])
    }
f()
...