add model performance to dichotomous risks

ihmeuw-msca · Feb 15, 2024 · 214e4b1 · 214e4b1
1 parent f5540af
commit 214e4b1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/bopforge/dichotomous_pipeline/functions.py b/src/bopforge/dichotomous_pipeline/functions.py
@@ -6,6 +6,7 @@
 from mrtool import MRBRT, CovFinder, LinearCovModel, MRData
 from pandas import DataFrame
 from scipy.stats import norm
+from limetr import get_aic, get_bic, get_rmse
 
 
 def get_signal_model(settings: dict, df: DataFrame) -> MRBRT:
@@ -257,6 +258,14 @@ def get_linear_model_summary(
     summary["pub_bias"] = int(pval < 0.05)
     summary["pub_bias_pval"] = float(pval)
 
+    summary["model_performance"] = {
+        "linear_model": {
+            "aic": float(get_aic(linear_model.lt)),
+            "bic": float(get_bic(linear_model.lt)),
+            "rmse": float(get_rmse(linear_model.lt)),
+        }
+    }
+
     return summary