R: modelrをつかってみるテスト [統計]
modelrとggplot2で回帰直線をひいてみたりするテストです。
コードlibrary(ggplot2) library(dplyr) library(modelr) data(iris) p1 <- ggplot(data = iris) + geom_point(mapping = aes(x = Sepal.Length, y = Sepal.Width, colour = Species)) print(p1) p2 <- p1 + geom_smooth(mapping = aes(x = Sepal.Length, y = Sepal.Width, colour = Species)) print(p2) spp <- levels(factor(iris$Species)) model <- lm(Sepal.Width ~ 0 + Sepal.Length + Species, data = iris) grid <- expand.grid(c(min(iris$Sepal.Length), max(iris$Sepal.Length)), spp) %>% rename(Sepal.Length = Var1, Species = Var2) %>% add_predictions(model) p3 <- p1 + geom_line(data = grid, aes(x = Sepal.Length, y = pred, colour = Species)) print(p3)
結果です。
p1
p2
p3
コメント 0