M/PhmdZddlZddlmZddlmZddlmZdZ dZ Gdd eZ Gd d eZ Gd d eZ dS)a Created on Mon Jul 26 08:34:59 2010 Author: josef-pktd changes: added offset and zero-inflated version of Poisson - kind of ok, need better test cases, - a nan in ZIP bse, need to check hessian calculations - found error in ZIP loglike - all tests pass with Issues ------ * If true model is not zero-inflated then numerical Hessian for ZIP has zeros for the inflation probability and is not invertible. -> hessian inverts and bse look ok if row and column are dropped, pinv also works * GenericMLE: still get somewhere (where?) "CacheWriteWarning: The attribute 'bse' cannot be overwritten" * bfgs is too fragile, does not come back * `nm` is slow but seems to work * need good start_params and their use in genericmle needs to be checked for consistency, set as attribute or method (called as attribute) * numerical hessian needs better scaling * check taking parts out of the loop, e.g. factorial(endog) could be precalculated N)stats) factorial)GenericLikelihoodModelcTtjtj||z S)Nnpmaxabsarr1arr2s \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/miscmodels/count.pymaxabsr$s 6"&%% & &&cZtjtj||z dz S)Nrr s r maxabsrelr's$ 6"&q)) * **rceZdZdZdZdZdS) PoissonGMLEafMaximum Likelihood Estimation of Poisson Model This is an example for generic MLE which has the same statistical model as discretemod.Poisson. Except for defining the negative log-likelihood method, all methods and results are generic. Gradients and Hessian and all resulting statistics are based on numerical differentiation. ctj|j|}|j}tj|||zz tjt |zS| Loglikelihood of Poisson model Parameters ---------- params : array_like The parameters of the model. Returns ------- The log likelihood of the model evaluated at `params` Notes ----- .. math:: \ln L=\sum_{i=1}^{n}\left[-\lambda_{i}+y_{i}x_{i}^{\prime}\beta-\ln y_{i}!\right] )rdotexogendogexplogr)selfparamsXBrs r nloglikeobszPoissonGMLE.nloglikeobs:sL"VDIv & & vbzzU2X%y/?/?(@(@@@rct|dst|j}|j}t jt j||}tj|dS)zOreturn frozen scipy.stats distribution with mu at estimated prediction resultr)loc) hasattr ValueErrorr#rrrrrpoisson)rrr#rmus rpredict_distributionz PoissonGMLE.predict_distributionOs\tX&& , [F]FtV,,--B=+++ +rN)__name__ __module__ __qualname____doc__r!r)rrrr,s?  AAA* , , , , ,rrc*eZdZdZdfd ZdZxZS)PoissonOffsetGMLEauMaximum Likelihood Estimation of Poisson Model This is an example for generic MLE which has the same statistical model as discretemod.Poisson but adds offset Except for defining the negative log-likelihood method, all methods and results are generic. Gradients and Hessian and all resulting statistics are based on numerical differentiation. Nnonec |1|jdkr |dddf}||_nd|_tj||fd|i|dS)Nrmissing)ndimraveloffsetsuper__init__rrrr7r4kwds __class__s rr9zPoissonOffsetGMLE.__init__ls|  {a$ ,,..DKKDK  g      rc|jtj|j|z}|j}tj|||zz tjt|z}|Sr)r7rrrrrrr)rrr rnlogliks rr!zPoissonOffsetGMLE.nloglikeobs|sW$[26$)V44 4 &**b(26)E2B2B+C+CCrNNr1r*r+r,r-r9r! __classcell__r<s@rr0r0_sV         rr0c*eZdZdZdfd ZdZxZS) PoissonZiGMLEaMaximum Likelihood Estimation of Poisson Model This is an example for generic MLE which has the same statistical model as discretemod.Poisson but adds offset and zero-inflation. Except for defining the negative log-likelihood method, all methods and results are generic. Gradients and Hessian and all resulting statistics are based on numerical differentiation. There are numerical problems if there is no zero-inflation. Nr1c d|_tj||f|dgd||1|jdkr |dddf}||_nd|_| t j|jdf|_ |j j d|_ t j t j|j df|_ |xj dz c_ dg|_dS)Nrzi)r4extra_params_namesr3r start_params)k_extrar8r9r5r6r7ronesnobsrshapenparamshstackrH cloneattrr:s rr9zPoissonZiGMLE.__init__s  3g$(6 3 3-1 3 3 3  {a$ ,,..DKKDK <1 ..DIyq) Irwt|'<'s rr!zPoissonZiGMLE.nloglikeobss"crc{Avbz***+[26$)T22 2 6!D&>>/BF2JJ.%(:RVIeDTDT=U=UU fTBFGE1H4E3E,F,F%FGGGqrr?r@rBs@rrDrDsV  ******6rrD)r-numpyrscipyr scipy.specialrstatsmodels.base.modelrrrrr0rDr.rrrXs:######999999'''+++ /,/,/,/,/,(/,/,/,f22222.222hDDDDD*DDDDDr