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)