fix: closes #213

rsquaredacademy · Jul 19, 2024 · 6bfef19 · 6bfef19
1 parent ce96c6a
commit 6bfef19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/R/ols-regress_compute.R b/R/ols-regress_compute.R
@@ -20,7 +20,7 @@ reg_comp <- function(formula, data, conf.level = 0.95, iterm, title = "model") {
   sbic       <- ols_sbic(model, model)
   prsq       <- ols_pred_rsq(model)
   n          <- length(anovam$Df)
-  mse        <- anovam$`Mean Sq`[n]
+  mse        <- mean(model$residuals ^ 2)
   error_df   <- anovam$Df[n]
   model_df   <- sum(anovam$Df) - error_df
   total_df   <- sum(anovam$Df)

diff --git a/tests/testthat/_snaps/regress-output.md b/tests/testthat/_snaps/regress-output.md
@@ -6,7 +6,7 @@
                                Model Summary                          
       ---------------------------------------------------------------
       R                       0.909       RMSE                 2.468 
-      R-Squared               0.827       MSE                  6.964 
+      R-Squared               0.827       MSE                  6.093 
       Adj. R-Squared          0.808       Coef. Var           13.135 
       Pred R-Squared          0.768       AIC                158.643 
       MAE                     1.907       SBC                165.972 
@@ -45,7 +45,7 @@
                                Model Summary                          
       ---------------------------------------------------------------
       R                       0.909       RMSE                 2.468 
-      R-Squared               0.827       MSE                  6.964 
+      R-Squared               0.827       MSE                  6.093 
       Adj. R-Squared          0.808       Coef. Var           13.135 
       Pred R-Squared          0.768       AIC                158.643 
       MAE                     1.907       SBC                165.972