diff --git a/yli/regress.py b/yli/regress.py index ba951ab..bd20806 100644 --- a/yli/regress.py +++ b/yli/regress.py @@ -145,10 +145,14 @@ class RegressionResult: """ Compute Bayes factor testing the hypothesis that the given beta is 0 Requires statsmodels regression + + term: Raw name (from statsmodels, available via RegressionResult.raw_result) of term to be tested """ + # FIXME: Allow specifying our renamed terms + # Get parameters required for AFBF - params = pd.Series({term.raw_name.replace('[', '_').replace(']', '_'): term.beta.point for term in self.terms.values()}) + params = pd.Series({raw_name.replace('[', '_').replace(']', '_'): beta for raw_name, beta in self.raw_result.params.items()}) cov = self.raw_result.cov_params() # Compute BF matrix