Comment tracer 3 modèles dans une figure dans R?

Question

Je suis nouveau avec R et j'ai 3 modèles pour ajustement mes données comme suit:

Modèle 1: y = a (x) + b

lm1 = lm(data$CBI ~ data$dNDVI) 

Modèle 2: y = a (x) 2 + b (x) + c

lm2 <- lm(CBI ~ dNDVI + I(dNDVI^2), data=data) 

Modèle 3: y = x (a | x | + b) -1

lm3 = nls(CBI ~ dNDVI*(a*abs(dNDVI) + b) - 1, start = c(a = 1.5, b = 2.7), data = data) 

maintenant, je voudrais tracer tous ces trois modèles en R mais je ne pouvais pas trouver le moyen de le faire, pouvez-vous s'il vous plaît m'aider? J'ai essayé avec les deux premiers modèles suivent et ça marche mais je ne sais pas comment ajouter le modèle 3 sur elle:

ggplot(data = data, aes(x = dNDVI, y = CBI)) + 
    geom_point() + 
    geom_smooth(method = lm, formula = y ~ x, size = 1, se = FALSE) + 
    geom_smooth(method = lm, formula = y ~ x + I(x^2), size = 1, se = FALSE) + 
    theme_bw() 

Je voudrais également ajouter une légende qui montre 3 couleurs ou types différents des lignes/courbes pour les 3 modèles. Pouvez-vous s'il vous plaît me guider comment le faire dans la figure?

Answer 1

utilisant l'iris comme un ensemble factice pour représenter les trois modèles:

new.dat <- data.frame(Sepal.Length=seq(min(iris$Sepal.Length), 
         max(iris$Sepal.Length), length.out=50)) #new data.frame to predict the fitted values for each model 

m1 <- lm(Petal.Length ~ Sepal.Length, iris) 
m2 <- lm(Petal.Length ~ Sepal.Length + I(Sepal.Length^2), data=iris) 
m3 <- nls(Petal.Length ~ Sepal.Length*(a*abs(Sepal.Length) + b) - 1, 
     start = c(a = 1.5, b = 2.7), data = iris) 

new.dat$m1.fitted <- predict(m1, new.dat) 
new.dat$m2.fitted <- predict(m2, new.dat) 
new.dat$m3.fitted <- predict(m3, new.dat) 

new.dat <- new.dat %>% gather(var, val, m1.fitted:m3.fitted) #stacked format of fitted data of three models (to automatically generate the legend in ggplot) 


ggplot(new.dat, aes(Sepal.Length, val, colour=var)) + 
     geom_line()