From ce4e61c3d5d1817c623b1f5aedc11ecfb5e1b19d Mon Sep 17 00:00:00 2001 From: connorschwartz <46463980+connorschwartz@users.noreply.github.com> Date: Mon, 4 May 2026 21:43:07 -0400 Subject: [PATCH] Add standard errors to MRA params output --- openavmkit/modeling.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/openavmkit/modeling.py b/openavmkit/modeling.py index 55d8641..e0a037c 100644 --- a/openavmkit/modeling.py +++ b/openavmkit/modeling.py @@ -5597,8 +5597,12 @@ def write_mra_params( csv_path = f"{outpath}/params.csv" params = model.fitted_model.params.copy() # pandas Series params = params.rename(index={"const": "intercept"}) # const -> intercept - - df_coef = params.to_frame(name="coefficient") + errors = model.fitted_model.bse.copy() + errors = errors.rename(index={"const": "intercept"}) + + df_params = params.to_frame(name="coefficient") + df_errors = errors.to_frame(name="error") + df_coef = pd.concat([df_params, df_errors], axis=1) df_coef.index.name = "variable" df_coef.to_csv(csv_path)