M/Ph`;ddlZddlmZmZmZddlmZddlm Z ddl m Z GddeZ e dd Z Gd d eZed d ZGddeZedd ZGddeZedd ZGddeZedd ZGddeZedd ZGdd eZGd!d"e ZdS)#N) rv_discretepoissonnbinom)gammaln) _lazywhere)GenericLikelihoodModelc0eZdZdZdZdZdZdZdZdS)genpoisson_p_genz%Generalized Poisson distribution c*|dk||kz|dkzS)Nrselfmualphaps b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/distributions/discrete.py _argcheckzgenpoisson_p_gen._argcheck saE5L)QU33c||dz z}tjtjddd||zz}tjtjdd||dz |zz}tj||dz tj|zz}||tj|zt |dzz||z zz}|S)N?r)npmaximum nextafterlogr) rxrrrmu_pa1a2logpmf_s r_logpmfzgenpoisson_p_gen._logpmfsa"f~ Z Q**A ,< = = Z Q**B"r'Q,> ? ?&**B"&**44Ar NWQV__4rBw>>rcVtj|||||SNrexpr!)rrrrrs r_pmfzgenpoisson_p_gen._pmfs$vdll1b%33444rc|Sr#r r s rmeanzgenpoisson_p_gen.means rc2d|||dz zzzdz}||z}|SNrr )rrrrdispersion_factorvars rr-zgenpoisson_p_gen.vars.a!e!44q8"$ rN __name__ __module__ __qualname____doc__rr!r&r(r-r rrr r si444555rr genpoisson_pzGeneralized Poisson)namelongnamecBeZdZdZdZdZdZdZdZdZ dZ d Z d S) zipoisson_genz'Zero Inflated Poisson distribution c*|dk|dkz|dkzSNrrr rrws rrzzipoisson_gen._argcheck+Q16"ad++rc t|dk|||fdtj|d|z tj| zzS)Nrctjd|z |tj|zzt|dzz |z SNr)rrr)rrr;s rz'zipoisson_gen._logpmf..0s=BF26NNQ^,K!!b&//-*,.-/rr)rrrr%rrrr;s rr!zzipoisson_gen._logpmf.sQ!q&1b!*//&b1fs %;!;<<>> >rcTtj||||Sr#r$rAs rr&zzipoisson_gen._pmf4s"vdll1b!,,---rcZ|t||d|z zzS)Nrr)rcdfrAs r_cdfzzipoisson_gen._cdf7s,7b>>>%%a((AE222rc||z d|z z }t||}t|tjr d|||k<ntj|r||krdS|S)NrrDr)rppf isinstancerndarrayisscalar)rqrr;q_modrs r_ppfzzipoisson_gen._ppf;stQ1q5! rNNN  u % % a $ $ Aa!eHH [^^ A3rcd|z |zSNrr r:s rr(zzipoisson_gen.meanFA|rcHd||zz}||||z}|SrQr()rrr;r,r-s rr-zzipoisson_gen.varIs-BJ 499R#3#33 rc8d|z tj||zSrQ)rmoment)rnrr;s r_momentzzipoisson_gen._momentNsA2....rN) r/r0r1r2rr!r&rFrOr(r-rXr rrr7r7(s,,,>>> ...333    /////rr7 zipoissonzZero Inflated Poissonc0eZdZdZdZdZdZdZdZdS)zigeneralizedpoisson_genz3Zero Inflated Generalized Poisson distribution c*|dk|dkz|dkzSr9r rrrrr;s rrz"zigeneralizedpoisson_gen._argcheckXr<rct|dk|||||fdtj|d|z t||||zzS)Nrcltjd|z t||||zSr?)rrr3logpmf)rrrrr;s rr@z2zigeneralizedpoisson_gen._logpmf..]s.bfR!Vnn&--aUA>>7?rr)rrrr3pmfrrrrrr;s rr!z zigeneralizedpoisson_gen._logpmf[si!q&1b%A"6??&b1f&**1b%;;&<"<==>> >rc Xtj||||||Sr#r$rbs rr&zzigeneralizedpoisson_gen._pmfb&vdll1b%A66777rcd|z |zSrQr r]s rr(zzigeneralizedpoisson_gen.meanerRrcn|dz }d|||zzzdz||zz}||||||z}|Sr*rTrrrrr;r,r-s rr-zzigeneralizedpoisson_gen.varhsM Eq0Q6R? 499R1#=#== rNr.r rrr[r[Usi,,,>>>888rr[ zigenpoissonz!Zero Inflated Generalized PoissoncHeZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) zinegativebinomial_genz=Zero Inflated Generalized Negative Binomial distribution c*|dk|dkz|dkzSr9r r]s rrz zinegativebinomial_gen._argcheckwr<rc||||\}}t|dk||||fdtj|d|z t j|||zzS)Nrc^tjd|z tj|||zSr?)rrrr`)rsrr;s rr@z0zinegativebinomial_gen._logpmf..}s(bfR!Vnn -1a00/1rr)convert_paramsrrrrrarrrrrr;rns rr!zzinegativebinomial_gen._logpmfzs{""2ua001!q&1aA,11&b1f *Q1--&.".//00 0rc Xtj||||||Sr#r$rbs rr&zzinegativebinomial_gen._pmfrdrct||||\}}|tj|||d|z zzSrQ)rorrErps rrFzzinegativebinomial_gen._cdfsA""2ua0016:aA&&!a%000rc||||\}}||z d|z z }tj|||}t|tjr d|||k<nt j|r||krdS|S)NrrrH)rorrIrJrrKrL) rrMrrrr;rnrNrs rrOzzinegativebinomial_gen._ppfs""2ua001Q1q5! Jua # # a $ $ Aa!eHH [^^ A3rcd|z |zSrQr r]s rr(zzinegativebinomial_gen.meanrRrcdd|||dz zzz||zz}||||||z}|SrQrTrgs rr-zzinegativebinomial_gen.varsDq1u 55B> 499R1#=#== rcn||||\}}d|z tj|||zSrQ)rorrV)rrWrrrr;rns rrXzzinegativebinomial_gen._moments:""2ua001Aq!Q////rc6d|z |d|z zz}|||zz }||fSNrr+r rrrrsizeprobs rroz%zinegativebinomial_gen.convert_params0EzB1I%tby!d|rN) r/r0r1r2rr!r&rFrOr(r-rXror rrrjrjts,,,000888111    000rrjzinegbinz+Zero Inflated Generalized Negative Binomialc*eZdZdZdZdZdZdZdS)truncatedpoisson_genz/Truncated Poisson discrete random variable c|dk|dkzSNrr rr truncations rrztruncatedpoisson_gen._argchecksaJ",--rc|dz|jfSrQbrs r _get_supportz!truncatedpoisson_gen._get_supportA~tv%%rcd}tttj|dzD]}|t j||z }tj|tj }|dk}tj||<|dk}tj d||z ||<t j |||z }|Sr9) rangeintrmaxrra full_likeinfnanrr`) rrrrrai log_1_m_pmflocr s rr!ztruncatedpoisson_gen._logpmfss26*--..233 & &A 7;q"%% %CCl300 Ag6 CAg6!c#h,// C.B''+5rcTtj||||Sr#r$)rrrrs rr&ztruncatedpoisson_gen._pmfs"vdll1b*55666rN)r/r0r1r2rrr!r&r rrrrsZ... &&&   77777rrtruncatedpoissonzTruncated Poissonc0eZdZdZdZdZdZdZdZdS)truncatednegbin_genzLTruncated Generalized Negative Binomial (NB-P) discrete random variable c|dk|dkzSrr rrrrrs rrztruncatednegbin_gen._argchecksaJ",--rc|dz|jfSrQrrs rrz truncatednegbin_gen._get_supportrrc||||\}}d}tttj|dzD]} |t j| ||z }tj|tj } |dk} tj | | <|dk} tj d|| z | | <t j |||| z } | Sr9) rorrrrrrarrrrr`) rrrrrrrzr{rarrrr s rr!ztruncatednegbin_gen._logpmfs((UA66 ds26*--..233 - -A 6:at,, ,CCl300 Ag6 CAg6!c#h,// C-4..<rc Xtj||||||Sr#r$)rrrrrrs rr&ztruncatednegbin_gen._pmfs&vdll1b%J??@@@rc6d|z |d|z zz}|||zz }||fSrxr rys rroz"truncatednegbin_gen.convert_paramsr|rN) r/r0r1r2rrr!r&ror rrrrsl...&&& AAArrtruncatednegbinz'Truncated Generalized Negative BinomialcjeZdZdZfdZdfd ZfdZdZddd d Zd Z d Z d Z dZ dZ xZS)DiscretizedCountunCount distribution based on discretized distribution Parameters ---------- distr : distribution instance d_offset : float Offset for integer interval, default is zero. The discrete random variable is ``y = floor(x + offset)`` where x is the continuous random variable. Warning: not verified for all methods. add_scale : bool If True (default), then the scale of the base distribution is added as parameter for the discrete distribution. The scale parameter is in the last position. kwds : keyword arguments The extra keyword arguments are used delegated to the ``__init__`` of the super class. Their usage has not been checked, e.g. currently the support of the distribution is assumed to be all non-negative integers. Notes ----- `loc` argument is currently not supported, scale is not available for discrete distributions in scipy. The scale parameter of the underlying continuous distribution is the last shape parameter in this DiscretizedCount distribution if ``add_scale`` is True. The implementation was based mainly on [1]_ and [2]_. However, many new discrete distributions have been developed based on the approach that we use here. Note, that in many cases authors reparameterize the distribution, while this class inherits the parameterization from the underlying continuous distribution. References ---------- .. [1] Chakraborty, Subrata, and Dhrubajyoti Chakravarty. "Discrete gamma distributions: Properties and parameter estimations." Communications in Statistics-Theory and Methods 41, no. 18 (2012): 3301-3324. .. [2] Alzaatreh, Ayman, Carl Lee, and Felix Famoye. 2012. “On the Discrete Analogues of Continuous Distributions.” Statistical Methodology 9 (6): 589–603. cTtt||Sr#)superr__new__)clsargskwds __class__s rrzDiscretizedCount.__new__"s#[#&&..s333rrTc ||_||_|j|_||_|j^t |jd|_|r/|d|jdzi|xjdz c_n(|r|ddid|_nd|_tj di|dS)N,shapesz, srrnrr ) distrd_offset _ctor_param add_scalerlensplitk_shapesupdater__init__)rrrrrrs rrzDiscretizedCount.__init__'s    ," < # 2 23 7 788DM # Xu|e';<=== "  " XsO,,, ! !   4     rc\t}|j|d<|S)Nr)r_updated_ctor_paramr)rdicrs rrz$DiscretizedCount._updated_ctor_param<s(gg))++zG  rcB|jr|d}|dd}nd}||fS)Nrr)r)rrscales r _unpack_argszDiscretizedCount._unpack_argsAs3 > HE9DDEU{rN)rz random_statec ||\}}|t|dd}tj|jj||||d|jz}|S)N_sizer)rrzr)rgetattrrtruncrrvsr)rrzrrrrvs r_rvszDiscretizedCount._rvsIst''-- e <4!,,D Xndjnd%d2>@@@m$%% rc|j}|jdkr ||jz}||\}}|j|g|Rd|i|j|dzg|Rd|iz }|S)Nrrr)rrrsfrrrrrrs rr&zDiscretizedCount._pmfRs  =A  DM!A''-- e UXa ,$ , , ,e , , UXa!e 0d 0 0 0% 0 01rc|j}||\}}|jdkr ||jz}|j|dzg|Rd|i}|SNrrr)rrrrErs rrFzDiscretizedCount._cdf]sc ''-- e =A  DM!A EIa!e 0d 0 0 0% 0 0rc|j}||\}}|jdkr ||jz}|j|dzg|Rd|i}|Sr)rrrrrs r_sfzDiscretizedCount._sfesc ''-- e =A  DM!A EHQU /T / / / / /rc|j}||\}}|j|g|Rd|i}|jdkr ||jz}t j|dz}|SNrrg?)rrrIrrfloorrrrrrqcrMs rrOzDiscretizedCount._ppfms ''-- e UYq -4 - - -u - - =A  dm#B HR9% & &rc|j}||\}}|j|g|Rd|i}|jdkr ||jz}t j|dz}|Sr)rrisfrrrrs r_isfzDiscretizedCount._isfwrr)rT)r/r0r1r2rrrrrr&rFrrOr __classcell__rs@rrrs,,\44444 !!!!!!*  $$   rrc8eZdZdZdfd ZdZd dZdZxZS) DiscretizedModelaexperimental model to fit discretized distribution Count models based on discretized distributions can be used to model data that is under- or over-dispersed relative to Poisson or that has heavier tails. Parameters ---------- endog : array_like, 1-D Univariate data for fitting the distribution. exog : None Explanatory variables are not supported. The ``exog`` argument is only included for consistency in the signature across models. distr : DiscretizedCount instance (required) Instance of a DiscretizedCount distribution. See Also -------- DiscretizedCount Examples -------- >>> from scipy import stats >>> from statsmodels.distributions.discrete import ( DiscretizedCount, DiscretizedModel) >>> dd = DiscretizedCount(stats.gamma) >>> mod = DiscretizedModel(y, distr=dd) >>> res = mod.fit() >>> probs = res.predict(which="probs", k_max=5) Ncl|tdt||||jdt ||jz |_d|_|j|_ d|_ |j|_ dtj |j z|_dS)Nexog is not supportedrrrg?) ValueErrorrr _init_keysappendrrdf_residdf_modelk_extra k_constantnparamsrones start_params)rendogexogrrs rrzDiscretizedModel.__init__s  455 5 E222 w'''E U^3  ~ ~ "'$,"7"77rc|}tj|jj|jg|R}|Sr#)rrrr&rsum)rparamsrlls rloglikezDiscretizedModel.loglikes=  VODJODJ6666 7 7vvxxrc|td|}|dkr'|jjtj|g|R}|Std)Nrprobsz+only which="probs" is currently implemented)rrrararange)rrrwhichk_maxrprs rpredictzDiscretizedModel.predicts`  455 5 G   % 0 084888BIJKK Krc|}|j|}|S)zCfrozen distribution instance of the discrete distribution. r)rrrrs r get_distrzDiscretizedModel.get_distrs D! r)NN)NNr) r/r0r1r2rrrrrrs@rrrs~@ 8 8 8 8 8 8 L L L Lrr)numpyr scipy.statsrrr scipy.specialrscipy._lib._utilrstatsmodels.base.modelrr r3r7rYr[rhrjr}rrrrrrr rrrs4444444444!!!!!!''''''999999{4 ^)>@@@ '/'/'/'/'/K'/'/'/T M{#: < < < {4('  0222 22222[222h " !z : < < <77777;777@('-?1DFFF+B&%+< 6888LLLLL{LLL^HHHHH-HHHHHr