ó áp7]c@sòddlZddlmZmZddlmZddlmZdefd„ƒYZ e ddd d ƒZ d efd „ƒYZ e dd d dƒZ defd„ƒYZ e ddd dƒZdefd„ƒYZeddd dƒZdS(iÿÿÿÿN(t rv_discretetnbinom(tgammaln(t _lazywheretgenpoisson_p_gencBs)eZdZd„Zd„Zd„ZRS(s%Generalized Poisson distribution cCs|dk||k@|dk@S(Ni((tselftmutalphatp((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyt _argcheck sc Cs³||d}tjtjddƒd||ƒ}tjtjddƒ||d|ƒ}tj|ƒ|dtj|ƒ}||tj|ƒt|dƒ||8}|S(Ngð?ii(tnptmaximumt nextaftertlogR( RtxRRRtmu_pta1ta2tlogpmf_((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyt_logpmf s &*$-cCstj|j||||ƒƒS(N(R texpR(RRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyt_pmfs(t__name__t __module__t__doc__R RR(((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyRs  tnamet genpoisson_ptlongnamesGeneralized Poissont zipoisson_gencBs)eZdZd„Zd„Zd„ZRS(s'Zero Inflated Poisson distribution cCs|dk|dk@|dk@S(Nii((RRtw((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR sc CsDt|dk|||fd„tj|d|tj| ƒƒƒS(NicSs4tjd|ƒ|tj|ƒt|dƒ|S(Ngð?(R R R(RRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyt#sgð?(RR R R(RRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR!scCstj|j|||ƒƒS(N(R RR(RRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR's(RRRR RR(((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyRs  t zipoissonsZero Inflated Poissontzigeneralizedpoisson_gencBs)eZdZd„Zd„Zd„ZRS(s3Zero Inflated Generalized Poisson distribution cCs|dk|dk@|dk@S(Nii((RRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR 0sc CsRt|dk|||||fd„tj|d|tj||||ƒƒƒS(NicSs'tjd|ƒtj||||ƒS(Ngð?(R R Rtlogpmf(RRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR5sgð?(RR R Rtpmf(RRRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR3scCs"tj|j|||||ƒƒS(N(R RR(RRRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR:s(RRRR RR(((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR -s  t zigenpoissons!Zero Inflated Generalized Poissontzinegativebinomial_gencBs2eZdZd„Zd„Zd„Zd„ZRS(s=Zero Inflated Generalized Negative Binomial distribution cCs|dk|dk@|dk@S(Nii((RRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR Csc Csg|j|||ƒ\}}t|dk||||fd„tj|d|tj|||ƒƒƒS(NicSs$tjd|ƒtj|||ƒS(Ngð?(R R RR!(RtsRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyRIsgð?(tconvert_paramsRR R RR"(RRRRRRR%((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyRFs cCs"tj|j|||||ƒƒS(N(R RR(RRRRRR((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyRNscCs.d||d|}|||}||fS(Ngð?i((RRRRtsizetprob((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR&Qs(RRRR RRR&(((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyR$@s    tzinegbins+Zero Inflated Generalized Negative Binomial(tnumpyR t scipy.statsRRt scipy.specialRtstatsmodels.compat.scipyRRRRRR R#R$R)(((sAlib/python2.7/site-packages/statsmodels/distributions/discrete.pyts