Show R^2 for regression models

Author: OndrejSlamecka

Project.toml

@@ -1,6 +1,6 @@
 name = "StatsModels"
 uuid = "3eaba693-59b7-5ba5-a881-562e759f1c8d"
-version = "0.6.29"
+version = "0.6.30"
 
 [deps]
 DataAPI = "9a962f9c-6df0-11e9-0e5d-c546b8b5ee8a"

src/StatsModels.jl

@@ -33,7 +33,7 @@ export
     SeqDiffCoding,
     HypothesisCoding,
     ContrastsCoding,
-    
+
     coefnames,
     setcontrasts!,
     formula,

src/statsmodel.jl

@@ -193,6 +193,12 @@ function StatsBase.coeftable(model::TableModels; kwargs...)
     ct
 end
 
+_show_fit_stats(io::IO, model::TableModels) = nothing
+
+function _show_fit_stats(io::IO, model::TableRegressionModel)
+    println("R²: ", round(r2(model), sigdigits=4))
+end
+
 # show function that delegates to coeftable
 function Base.show(io::IO, model::TableModels)
     println(io, typeof(model))
@@ -202,6 +208,7 @@ function Base.show(io::IO, model::TableModels)
     try
         println(io,"Coefficients:")
         show(io, coeftable(model))
+        _show_fit_stats(io, model)
     catch e
         if isa(e, MethodError) || isa(e, ErrorException) && occursin("coeftable is not defined", e.msg)
             show(io, model.model)