Ajuste de modelos lineares e mistos no ambiente R

09 e 10 de Outubro de 2014 - Piracicaba - SP
Prof. Dr. Walmes M. Zeviani
Escola Superior de Agricultura “Luiz de Queiroz” - USP
Lab. de Estatística e Geoinformação - LEG
Pós Graduação em Genética e Melhoramento de Plantas Departamento de Estatística - UFPR

Ajuste de modelo de regressão linear

##-----------------------------------------------------------------------------
## Definições da sessão, pacotes a serem usados.

pkg <- c("lattice", "latticeExtra", "reshape", "car", "alr3",
         "plyr", "wzRfun")
sapply(pkg, library, character.only=TRUE, logical.return=TRUE)
##      lattice latticeExtra      reshape          car         alr3         plyr 
##         TRUE         TRUE         TRUE         TRUE         TRUE         TRUE 
##       wzRfun 
##         TRUE
source("lattice_setup.R")

##-----------------------------------------------------------------------------
## Informações sobre as versões dos pacotes.

sessionInfo()
## R version 3.1.1 (2014-07-10)
## Platform: i686-pc-linux-gnu (32-bit)
## 
## locale:
##  [1] LC_CTYPE=en_US.UTF-8       LC_NUMERIC=C               LC_TIME=pt_BR.UTF-8       
##  [4] LC_COLLATE=en_US.UTF-8     LC_MONETARY=pt_BR.UTF-8    LC_MESSAGES=en_US.UTF-8   
##  [7] LC_PAPER=pt_BR.UTF-8       LC_NAME=C                  LC_ADDRESS=C              
## [10] LC_TELEPHONE=C             LC_MEASUREMENT=pt_BR.UTF-8 LC_IDENTIFICATION=C       
## 
## attached base packages:
## [1] stats     graphics  grDevices utils     datasets  base     
## 
## other attached packages:
## [1] wzRfun_0.1          plyr_1.8.1          alr3_2.0.5          car_2.0-21         
## [5] reshape_0.8.5       latticeExtra_0.6-26 RColorBrewer_1.0-5  lattice_0.20-29    
## [9] knitr_1.6          
## 
## loaded via a namespace (and not attached):
## [1] evaluate_0.5.5 formatR_0.10   grid_3.1.1     MASS_7.3-34    methods_3.1.1 
## [6] nnet_7.3-8     Rcpp_0.11.2    stringr_0.6.2  tools_3.1.1
## obs: Para instalar um pacote faça:
## install.packages("nome_do_pacote", dependencies=TRUE)

Distância para parada em função da velocidade

##-----------------------------------------------------------------------------
## Explorar os dados.

## Estrutura do Arquivo.
str(cars)
## 'data.frame':    50 obs. of  2 variables:
##  $ speed: num  4 4 7 7 8 9 10 10 10 11 ...
##  $ dist : num  2 10 4 22 16 10 18 26 34 17 ...
## Visualização.
xyplot(dist~speed, data=cars, type=c("p","smooth"))

plot of chunk unnamed-chunk-3

##-----------------------------------------------------------------------------
## Estimação por mínimos quadrados via solução matricial.

X <- cbind(b0=1, b1=cars$speed)
head(X)
##      b0 b1
## [1,]  1  4
## [2,]  1  4
## [3,]  1  7
## [4,]  1  7
## [5,]  1  8
## [6,]  1  9
y <- cbind(cars$dist)

## Estimativa dos parâmetros.
solve(t(X)%*%X)%*%t(X)%*%y
##       [,1]
## b0 -17.579
## b1   3.932
##-----------------------------------------------------------------------------
## Usando a lm().

## y ~ Normal(modelo linear, sigma²)
## modelo linear: b0+b1*x.

m0 <- lm(dist~speed, data=cars)
## m0 <- lm(dist~0+speed, data=cars)

c0 <- coef(m0)
c0
## (Intercept)       speed 
##     -17.579       3.932
## Sobrepondo ajuste às observações.
xyplot(dist~speed, data=cars, xlim=c(0,NA), ylim=c(c0[1],NA))+
    layer(panel.abline(a=c0[1], b=c0[2], col=2))

plot of chunk unnamed-chunk-3

## Diagnóstico.
par(mfrow=c(2,2)); plot(m0); layout(1)

plot of chunk unnamed-chunk-3

## Falta de ajuste. Extender para um modelo que permita curvatura da
## função.

##-----------------------------------------------------------------------------
## Ajustar polinômio de segundo grau.

## modelo linear: b0+b1*x+b2*x^2.
m1 <- lm(dist~speed+I(speed^2), data=cars)

## Diagnóstico.
par(mfrow=c(2,2)); plot(m1); layout(1)

plot of chunk unnamed-chunk-3

## Mostra uma possível relação média~variância. 

## Testa o abandono do termo extra por meio da mudança na soma de
## quadrados.
anova(m1, m0)
## Analysis of Variance Table
## 
## Model 1: dist ~ speed + I(speed^2)
## Model 2: dist ~ speed
##   Res.Df   RSS Df Sum of Sq   F Pr(>F)
## 1     47 10825                        
## 2     48 11354 -1      -529 2.3   0.14
summary(m1)
## 
## Call:
## lm(formula = dist ~ speed + I(speed^2), data = cars)
## 
## Residuals:
##    Min     1Q Median     3Q    Max 
## -28.72  -9.18  -3.19   4.63  45.15 
## 
## Coefficients:
##             Estimate Std. Error t value Pr(>|t|)
## (Intercept)    2.470     14.817    0.17     0.87
## speed          0.913      2.034    0.45     0.66
## I(speed^2)     0.100      0.066    1.52     0.14
## 
## Residual standard error: 15.2 on 47 degrees of freedom
## Multiple R-squared:  0.667,  Adjusted R-squared:  0.653 
## F-statistic: 47.1 on 2 and 47 DF,  p-value: 5.85e-12
## Verifica se cabe uma tranformação.
MASS::boxcox(m1); abline(v=0.5, col=2)

plot of chunk unnamed-chunk-3

## Usar lambda=0.5 como valor para transformar.
xyplot(sqrt(dist)~speed, data=cars)

plot of chunk unnamed-chunk-3

##-----------------------------------------------------------------------------
## Com a variável transformada.

## Modelo quadrático com transformação na resposta.
m2 <- lm(sqrt(dist)~speed+I(speed^2), data=cars)

## Diagnóstico.
par(mfrow=c(2,2)); plot(m2); layout(1)

plot of chunk unnamed-chunk-3

## Teste para as estimativas, h0: beta==0.
summary(m2)
## 
## Call:
## lm(formula = sqrt(dist) ~ speed + I(speed^2), data = cars)
## 
## Residuals:
##    Min     1Q Median     3Q    Max 
## -2.073 -0.726 -0.183  0.637  3.116 
## 
## Coefficients:
##             Estimate Std. Error t value Pr(>|t|)  
## (Intercept)  0.99034    1.08671    0.91    0.367  
## speed        0.36559    0.14919    2.45    0.018 *
## I(speed^2)  -0.00143    0.00484   -0.30    0.769  
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## 
## Residual standard error: 1.11 on 47 degrees of freedom
## Multiple R-squared:  0.71,   Adjusted R-squared:  0.698 
## F-statistic: 57.5 on 2 and 47 DF,  p-value: 2.33e-13
##-----------------------------------------------------------------------------
## Volta pro modelo mais simples.

## Modelo que abandona b2, ou seja, b2==0.
m3 <- lm(sqrt(dist)~speed, data=cars)

summary(m3)
## 
## Call:
## lm(formula = sqrt(dist) ~ speed, data = cars)
## 
## Residuals:
##    Min     1Q Median     3Q    Max 
## -2.068 -0.698 -0.180  0.591  3.153 
## 
## Coefficients:
##             Estimate Std. Error t value Pr(>|t|)    
## (Intercept)   1.2771     0.4844    2.64    0.011 *  
## speed         0.3224     0.0298   10.83  1.8e-14 ***
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## 
## Residual standard error: 1.1 on 48 degrees of freedom
## Multiple R-squared:  0.709,  Adjusted R-squared:  0.703 
## F-statistic:  117 on 1 and 48 DF,  p-value: 1.77e-14
## Teste para o abandono de b2 (o mesmo que o t do summary).
anova(m3, m2)
## Analysis of Variance Table
## 
## Model 1: sqrt(dist) ~ speed
## Model 2: sqrt(dist) ~ speed + I(speed^2)
##   Res.Df  RSS Df Sum of Sq    F Pr(>F)
## 1     48 58.3                         
## 2     47 58.2  1     0.108 0.09   0.77
## Diagnóstico para o modelo final.
par(mfrow=c(2,2)); plot(m3); layout(1)

plot of chunk unnamed-chunk-3

##-----------------------------------------------------------------------------
## Representa os resultados.

## Gráfico do modelo final.
c3 <- coef(m3)
xyplot(sqrt(dist)~speed, data=cars)+
    layer(panel.abline(a=c3[1], b=c3[2], col=2))

plot of chunk unnamed-chunk-3

## Na escala natural.
xyplot(dist~speed, data=cars, xlim=c(0,NA), ylim=c(0,NA))+
    layer(panel.curve((c3[1]+c3[2]*x)^2, col=2))

plot of chunk unnamed-chunk-3

##-----------------------------------------------------------------------------
## Com bandas de confiança.

pred <- data.frame(speed=seq(0, 30, length.out=30))
pred <- cbind(pred, predict(m3, newdata=pred, interval="confidence"))
str(pred)
## 'data.frame':    30 obs. of  4 variables:
##  $ speed: num  0 1.03 2.07 3.1 4.14 ...
##  $ fit  : num  1.28 1.61 1.94 2.28 2.61 ...
##  $ lwr  : num  0.303 0.695 1.086 1.477 1.867 ...
##  $ upr  : num  2.25 2.53 2.8 3.08 3.35 ...
xyplot(sqrt(dist)~speed, data=cars)+
    as.layer(xyplot(fit~speed, data=pred, type="l",
                    ly=pred$lwr, uy=pred$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))

plot of chunk unnamed-chunk-3

##-----------------------------------------------------------------------------
## Na escala original.

i <- c("fit","lwr","upr")
pred[,i] <- pred[,i]^2

xyplot(dist~speed, data=cars, xlim=c(0,NA), ylim=c(0,NA))+
    as.layer(xyplot(fit~speed, data=pred, type="l",
                    ly=pred$lwr, uy=pred$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))

plot of chunk unnamed-chunk-3


Ganho de peso em perus em função da metionina na dieta

##-----------------------------------------------------------------------------
## Ganho de peso de perus em função de metionina na dieta.

str(turk0)
## 'data.frame':    35 obs. of  2 variables:
##  $ A   : num  0 0 0 0 0 0 0 0 0 0 ...
##  $ Gain: int  644 631 661 624 633 610 615 605 608 599 ...
## help(turk0, help_type="html")

## A
##     Amount of methionine supplement (percent of diet)
## Gain
##     Pen weight increase (g)

## Diagrama de dispersão com linha de tendência.
xyplot(Gain~A, data=turk0, type=c("p","smooth"),
       xlab="Metionina (% da dieta)",
       ylab="Ganho de peso (g)")

plot of chunk unnamed-chunk-4

##-----------------------------------------------------------------------------
## Ajuste do modelo.

## Ajuste do modelo de regressão linear simples: b0+b1*x+b2*x^2;
m0 <- lm(Gain~A, turk0)

## Observados vs ajustados.
xyplot(Gain~A, data=turk0,
       xlab="Metionina (% da dieta)",
       ylab="Ganho de peso (g)")+
    layer(panel.abline(m0))

plot of chunk unnamed-chunk-4

## Resíduos.
par(mfrow=c(2,2)); plot(m0); layout(1)

plot of chunk unnamed-chunk-4

## Indiscutível falta de ajuste. Considerar um modelo que permita
## curvatura na função.

##-----------------------------------------------------------------------------
## Ajuste do modelo saturado (considerando A como fator). Considerar A
## como fator ocupa o mesmo espaço vetorial que ajustar um polinômio de
## grau k-1, em que k é o número de níveis. A equação da reta é o
## polinômio de grau um.

turk0$Afat <- factor(turk0$A)
m1 <- lm(Gain~poly(A, nlevels(Afat)-1), turk0)
anova(m1)
## Analysis of Variance Table
## 
## Response: Gain
##                            Df Sum Sq Mean Sq F value Pr(>F)    
## poly(A, nlevels(Afat) - 1)  5 150041   30008    88.6 <2e-16 ***
## Residuals                  29   9824     339                   
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
m1 <- lm(Gain~Afat, turk0)
anova(m1)
## Analysis of Variance Table
## 
## Response: Gain
##           Df Sum Sq Mean Sq F value Pr(>F)    
## Afat       5 150041   30008    88.6 <2e-16 ***
## Residuals 29   9824     339                   
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## Note que embora os modelos tenham especificações diferentes o quadro
## de anova é o mesmo. Isso porque ambos exploram o mesmo espaço
## vetorial de k-1 dimensões, apenas os vetores que formam a base desse
## espaço é que são diferentes. O conjunto de vetores que formam a base
## são as colunas da matriz do modelo, X.

## Resíduos.
par(mfrow=c(2,2)); plot(m1); layout(1)

plot of chunk unnamed-chunk-4

## Resíduos ok.

##-----------------------------------------------------------------------------
## Médias amostrais e os valores preditos.

## Médias amostrais.
aggregate(Gain~A, data=turk0, mean)
##      A  Gain
## 1 0.00 623.0
## 2 0.04 668.4
## 3 0.10 715.6
## 4 0.16 732.0
## 5 0.28 794.0
## 6 0.44 785.4
## Valores preditos pelo modelo saturado.
predict(m1, newdata=list(Afat=levels(turk0$Afat)))
##     1     2     3     4     5     6 
## 623.0 668.4 715.6 732.0 794.0 785.4
## Valores preditos pelo modelo m0
predict(m0, newdata=list(A=unique(turk0$A)))
##     1     2     3     4     5     6 
## 648.5 664.3 687.9 711.5 758.8 821.9
##-----------------------------------------------------------------------------
## Teste de falta de ajuste para o modelo m0.

anova(m0, m1)
## Analysis of Variance Table
## 
## Model 1: Gain ~ A
## Model 2: Gain ~ Afat
##   Res.Df   RSS Df Sum of Sq    F  Pr(>F)    
## 1     33 35176                              
## 2     29  9824  4     25353 18.7 1.1e-07 ***
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## Não passou na falta de ajuste conforme já se havia antecipado.

##-----------------------------------------------------------------------------
## Ajuste do modelo de segundo grau: b0+b1*x+b2*x^2;

m2 <- lm(Gain~A+I(A^2), turk0)
## m2 <- lm(Gain~poly(A, 2), turk0)

## Resíduos.
par(mfrow=c(2,2)); plot(m2); layout(1)

plot of chunk unnamed-chunk-4

##-----------------------------------------------------------------------------
## Teste de falta de ajuste para o modelo m2.

anova(m1, m2)
## Analysis of Variance Table
## 
## Model 1: Gain ~ Afat
## Model 2: Gain ~ A + I(A^2)
##   Res.Df   RSS Df Sum of Sq    F Pr(>F)
## 1     29  9824                         
## 2     32 11340 -3     -1516 1.49   0.24
## Model 1: Gain ~ Afat
## Model 2: Gain ~ A + I(A^2)
##   Res.Df     RSS Df Sum of Sq     F Pr(>F)
## 1     29  9823.6                          
## 2     32 11339.9 -3   -1516.2 1.492 0.2374

## Não apresentou falta de ajuste. Então é um modelo que pode ser mantido.

##-----------------------------------------------------------------------------
## Gráfico dos valores preditos com bandas de confiança.

## range(turk0$A)
pred2 <- data.frame(A=seq(0, 0.45, by=0.025))
a <- predict(m2, newdata=pred2, interval="confidence")
pred2 <- cbind(pred2, a)
str(pred2)
## 'data.frame':    19 obs. of  4 variables:
##  $ A  : num  0 0.025 0.05 0.075 0.1 0.125 0.15 0.175 0.2 0.225 ...
##  $ fit: num  626 649 670 690 708 ...
##  $ lwr: num  615 640 663 682 700 ...
##  $ upr: num  636 657 678 698 717 ...
xyplot(Gain~A, data=turk0,
     xlab="Metionina (% da dieta)",
     ylab="Ganho de peso (g)")+
    as.layer(xyplot(fit~A, data=pred2, type="l",
                    ly=pred2$lwr, uy=pred2$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))

plot of chunk unnamed-chunk-4

## Sobrepondo as estimativas de médias do modelo m1.
pred1 <- data.frame(Afat=levels(turk0$Afat))
a <- predict(m1, newdata=pred1, interval="confidence")
pred1 <- cbind(pred1, a)
pred1$A <- as.numeric(as.character(pred1$Afat))
## pred1

xyplot(Gain~A, data=turk0,
     xlab="Metionina (% da dieta)",
     ylab="Ganho de peso (g)")+
    as.layer(xyplot(fit~A, data=pred2, type="l",
                    ly=pred2$lwr, uy=pred2$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))+
    as.layer(xyplot(fit~(A+0.005), data=pred1, pch=19,
                    ly=pred1$lwr, uy=pred1$upr, cty="bars",
                    prepanel=prepanel.cbH, panel=panel.cbH))

plot of chunk unnamed-chunk-4

##-----------------------------------------------------------------------------
## Uso de pesos no ajuste do modelo de regressão.

## Imagine que não tem os dados completos, apenas as médias e
## frequências.

tu <- ddply(turk0, .(A), summarise,
            mGain=mean(Gain), nGain=length(Gain))
tu
##      A mGain nGain
## 1 0.00 623.0    10
## 2 0.04 668.4     5
## 3 0.10 715.6     5
## 4 0.16 732.0     5
## 5 0.28 794.0     5
## 6 0.44 785.4     5
m3 <- lm(mGain~A+I(A^2), data=tu, weights=nGain)
summary(m3)
## 
## Call:
## lm(formula = mGain ~ A + I(A^2), data = tu, weights = nGain)
## 
## Weighted Residuals:
##      1      2      3      4      5      6 
##  -8.04  14.44  16.17 -29.04  11.61  -1.82 
## 
## Coefficients:
##             Estimate Std. Error t value Pr(>|t|)    
## (Intercept)   625.54       6.24  100.20  2.2e-06 ***
## A             964.47      86.76   11.12   0.0016 ** 
## I(A^2)      -1362.07     198.34   -6.87   0.0063 ** 
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## 
## Residual standard error: 22.5 on 3 degrees of freedom
## Multiple R-squared:  0.99,   Adjusted R-squared:  0.983 
## F-statistic:  147 on 2 and 3 DF,  p-value: 0.00102
summary(m2)
## 
## Call:
## lm(formula = Gain ~ A + I(A^2), data = turk0)
## 
## Residuals:
##    Min     1Q Median     3Q    Max 
## -32.99 -16.54   2.19  12.79  36.06 
## 
## Coefficients:
##             Estimate Std. Error t value Pr(>|t|)    
## (Intercept)   625.54       5.23   119.7  < 2e-16 ***
## A             964.47      72.65    13.3  1.5e-14 ***
## I(A^2)      -1362.07     166.08    -8.2  2.3e-09 ***
## ---
## Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1
## 
## Residual standard error: 18.8 on 32 degrees of freedom
## Multiple R-squared:  0.929,  Adjusted R-squared:  0.925 
## F-statistic:  210 on 2 and 32 DF,  p-value: <2e-16
## Observe as estimativas dos parâmetros. Por que o valor pontual é
## igual nos dois modelos? Por que o erro padrão é maior com o uso das
## médias? Por que o R² é maior com o uso das médias? O que iria
## acontecer se não fossem usados os pesos?

##-----------------------------------------------------------------------------
## Sobrepondo os valores preditos com bandas de confiança obtido com o
## ajuste com as médias.

pred3 <- data.frame(A=seq(0, 0.45, by=0.025))
a <- predict(m3, newdata=pred3, interval="confidence")
pred3 <- cbind(pred3, a)
## pred3

xyplot(Gain~A, data=turk0,
     xlab="Metionina (% da dieta)",
     ylab="Ganho de peso (g)")+
    as.layer(xyplot(fit~A, data=pred2, type="l",
                    ly=pred2$lwr, uy=pred2$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))+
    as.layer(xyplot(fit~A, data=pred3, type="l",
                    ly=pred3$lwr, uy=pred3$upr, cty="bands",
                    prepanel=prepanel.cbH, panel=panel.cbH))

plot of chunk unnamed-chunk-4