B ¤ٹمZام @sˆddlZddlZddlZddlmZmZmZGdd„deƒZ dd„Z dd„Zd d „Ze dkگr„ddlmZddlmZejj ،Zeejejd چZejZded<e ee، ،Ze ee، ،Z ejj! ،Ze "ejdd…df،Z#ejdd…dd…fZej$j%edddچZej$j%edddچZej$j%edddچZej&edd…ddddddddddg fddچZe 'ej،Zej(eeej) *،dچ ،Z+dS) éN)عget_function_nameع iteritemsع getargspecc@sVeZdZdZiZiZdd„Zdd„Zdd„Zdd „Z d d„Z dd d„Zedd„ƒZ dS)عResultsWrapperz‚ Class which wraps a statsmodels estimation Results class and steps in to reattach metadata to results (if available) cCs||_|j|_dS)N)ع_resultsع__doc__)عselfعresults©r ْ7lib/python3.7/site-packages/statsmodels/base/wrapper.pyع__init__szResultsWrapper.__init__cCsdd„t|jƒDƒS)NcSsg|]}|‘qSr r )ع.0عxr r rْ sz*ResultsWrapper.__dir__..)عdirr)rr r rع__dir__szResultsWrapper.__dir__cs®‡fdd„}y|dƒ}Wntk r,YnXy||ƒStk rJYnXt||ƒ}|jj}ˆj |،}|rکt|tƒrک|j||df|dd…‍ژ}n|rھ|j||dچ}|S)Ncst ˆ|،S)N)عobjectع__getattribute__)عname)rr rعsz1ResultsWrapper.__getattribute__..rré)عhow) عAttributeErrorعgetattrعmodelعdataع_wrap_attrsعgetع isinstanceعtupleعwrap_output)rعattrrr عobjrrr )rrrs" zResultsWrapper.__getattribute__cCs|jS)N)ع__dict__)rr r rع__getstate__-szResultsWrapper.__getstate__cCs|j |،dS)N)r#عupdate)rZdict_r r rع__setstate__1szResultsWrapper.__setstate__FcCs&ddlm}|r| ،|||ƒdS)aésave a pickle of this instance Parameters ---------- fname : string or filehandle fname can be a string to a file path or filename, or a filehandle. remove_data : bool If False (default), then the instance is pickled without changes. If True, then all arrays with length nobs are set to None before pickling. See the remove_data method. In some cases not all arrays will be set to None. r)عsave_pickleN)عstatsmodels.iolib.smpickler'عremove_data)rعfnamer)r'r r rعsave5szResultsWrapper.savecCsddlm}||ƒS)Nr)عload_pickle)r(r,)عclsr*r,r r rعloadJszResultsWrapper.loadN)F)ع__name__ع __module__ع__qualname__rrع _wrap_methodsrrrr$r&r+عclassmethodr.r r r rrs rcGs i}x|D]}| |،q W|S)N)r%)Zdictsعresultعdr r rعunion_dictsPs r6csZt ˆ،‡‡fdd„ƒ}tˆƒ}tj|d|d|ddچ}tˆƒ}d|||jf|_|S)Ncsjt |d،}|jj}ˆrJtˆtƒrJ| ˆ|f|‍|ژˆdˆdd…،}nˆrf| ˆ|f|‍|ژˆ،}|S)Nrrr)rrrrrrr )rعargsعkwargsr rr")عfuncrr rعwrapperXs(zmake_wrapper..wrapperrré)Zvarargsعdefaultsz%s%s %s)ع functoolsعwrapsrعinspectZ formatargspecrr)r9rr:ZargspecZ formattedZ func_namer )r9rrعmake_wrapperWs r@cCsHxBt|jƒD]4\}}t||ƒs qt||ƒ}t||ƒ}t|||ƒqWdS)N)rr2عhasattrrr@عsetattr)عklassZwrappingعmethrr9r:r r rعpopulate_wrapperms rEع__main__)ع DataFrame)عcolumnsgً?Z interceptéےےےےT)عcolZdroprér;ééééé ééF)Zprepend)Zfamily),r?r=ZnumpyZnpZstatsmodels.compat.pythonrrrrrr6r@rEr/Zstatsmodels.apiZapiZsmZpandasrGZdatasetsZlongleyr.rZexogZ exog_nameZdfZendogعyZOLSZfitZ olsresultZRLMZ rlmresultZwfsعlogعoffsetZtoolsZcategoricalZadd_constantعroundZGLMZfamiliesZPoissonعmodr r r rعs2I .