From f8e56d96b15d59e3d3f65968f663cfd243f8caf9 Mon Sep 17 00:00:00 2001 From: RunasSudo Date: Fri, 2 Dec 2022 12:54:03 +1100 Subject: [PATCH] Add "method" parameter for yli.regress --- yli/regress.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/yli/regress.py b/yli/regress.py index 0e41bea..1769381 100644 --- a/yli/regress.py +++ b/yli/regress.py @@ -456,7 +456,7 @@ def regress( nan_policy='warn', model_kwargs=None, fit_kwargs=None, family=None, # common model_kwargs - cov_type=None, maxiter=None, start_params=None, # common fit_kwargs + cov_type=None, method=None, maxiter=None, start_params=None, # common fit_kwargs bool_baselevels=False, exp=None, _dmatrices=None, ): @@ -479,6 +479,7 @@ def regress( :type fit_kwargs: dict :param family: See statsmodels *GLM* constructor :param cov_type: See statsmodels *model.fit* + :param method: See statsmodels *model.fit* :param maxiter: See statsmodels *model.fit* :param start_params: See statsmodels *model.fit* :param bool_baselevels: Show reference categories for boolean independent variables even if reference category is *False* @@ -534,6 +535,8 @@ def regress( fit_kwargs = {} if cov_type is not None: fit_kwargs['cov_type'] = cov_type + if method is not None: + fit_kwargs['method'] = method if maxiter is not None: fit_kwargs['maxiter'] = maxiter if start_params is not None: