From 214e4b154cb8007eb305c3315418a8408d138920 Mon Sep 17 00:00:00 2001
From: zhengp0 <zhengp@uw.edu>
Date: Thu, 15 Feb 2024 14:55:13 -0800
Subject: [PATCH] add model performance to dichotomous risks

---
 src/bopforge/dichotomous_pipeline/functions.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/bopforge/dichotomous_pipeline/functions.py b/src/bopforge/dichotomous_pipeline/functions.py
index befa41b..660e70d 100644
--- 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