quick recipe for StatsModels

Right now only works for the bivariate case
JuliaPlots · Jan 30, 2020 · d05cdf3 · d05cdf3
1 parent a43fe45
commit d05cdf3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/StatsPlots.jl b/src/StatsPlots.jl
@@ -37,6 +37,7 @@ include("violin.jl")
 include("hist.jl")
 include("marginalhist.jl")
 include("bar.jl")
+include("statsmodels.jl")
 include("dendrogram.jl")
 include("andrews.jl")
 include("ordinations.jl")

diff --git a/src/statsmodels.jl b/src/statsmodels.jl
@@ -0,0 +1,8 @@
+# Note both predict and RegressionModel are defined in StatsBase
+@recipe function f(mod::RegressionModel)
+    newx = [ones(200) range(extrema(mod.model.pp.X[:,2])..., length = 200)]
+    newy, l, u = predict(mod, newx, interval = :confidence)
+    ribbon := (vec(u)-newy, newy-vec(l))
+    label --> "model"
+    newx[:,2], newy
+end