RunasSudo/scipy-yli
1
Fork 0
Code Issues 1 Pull Requests Releases Activity

Add "method" parameter for yli.regress

Browse Source
This commit is contained in:
RunasSudo 2022-12-02 12:54:03 +11:00
parent a48e96e96c
commit f8e56d96b1
Signed by: RunasSudo
GPG Key ID: 7234E476BF21C61A
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yli/regress.py
View File

@ -456,7 +456,7 @@ def regress(
nan_policy='warn', nan_policy='warn',
model_kwargs=None, fit_kwargs=None, model_kwargs=None, fit_kwargs=None,
family=None, # common model_kwargs 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, bool_baselevels=False, exp=None,
_dmatrices=None, _dmatrices=None,
): ):
@ -479,6 +479,7 @@ def regress(
:type fit_kwargs: dict :type fit_kwargs: dict
:param family: See statsmodels *GLM* constructor :param family: See statsmodels *GLM* constructor
:param cov_type: See statsmodels *model.fit* :param cov_type: See statsmodels *model.fit*
:param method: See statsmodels *model.fit*
:param maxiter: See statsmodels *model.fit* :param maxiter: See statsmodels *model.fit*
:param start_params: 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* :param bool_baselevels: Show reference categories for boolean independent variables even if reference category is *False*
@ -534,6 +535,8 @@ def regress(
fit_kwargs = {} fit_kwargs = {}
if cov_type is not None: if cov_type is not None:
fit_kwargs['cov_type'] = cov_type fit_kwargs['cov_type'] = cov_type
if method is not None:
fit_kwargs['method'] = method
if maxiter is not None: if maxiter is not None:
fit_kwargs['maxiter'] = maxiter fit_kwargs['maxiter'] = maxiter
if start_params is not None: if start_params is not None: