M/Ph<dZddlZddlZddlmZmZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZddlmZmZmZddlmZd d lmZd Zd Zd ezdgzZ eeZ!ee Z"GddZ#GddZ$Gdde#e$Z%Gdde#Z&Gdde$Z'ej(dZ)dZ*dZ+eiej,dgfeiej,dgfeiej,e+e)gfeiej,e+e)e*gfedd iej,e+e)e*gfeiej,e+e)e*gfedd iej,e+e)e*gfeiej,e)gfeiej,e)e*gfedd iej,e)e*gfeiej,e)e*gfeddiej,e)e*gfeiej,e)e*gfeddiej,e)e*gfedd iej,e)gfgZ-eeeeeeeegZ.d!Z/e/\Z0Z1ej23d"e-d#Z4dS)$zK Created on Sat Nov 13 12:48:37 2021 Author: Josef Perktod License: BSD-3 N)assert_allclose assert_equal) add_constant)PredictionResultsMonotonic) BinaryModelLogitProbitPoissonNegativeBinomialNegativeBinomialPGeneralizedPoisson)ZeroInflatedPoissonZeroInflatedNegativeBinomialPZeroInflatedGeneralizedPoisson)DATA)results_predictzaget aget2 educyr actlim totchrdocviszprivate medicaidconstc eZdZdZdZdZdS) CheckPredictc |j}|j}t|jdd}td|jdz d}t |j||j||jt |j||j|d|jztj |j}tj |j}t |dtj |d|jdS)Nrrrtol) res1res2slicek_inflrparamsrbsenpasarrayexp)selfrrsl1sl2params1params2s g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/discrete/tests/test_predict.py test_basiczCheckPredict.test_basic/syyDKT**Aq)400 C($+c*:KKKK tx}2 >JJJJ*T[))*T[)) RVGBK%8%8tyIIIIIIc|j}|j}tjtd}|j}|j|fi|j}t|j |dj ddt|j |dj dddt|tr|d}t|d|dj dddt|d |d j ddd|d d}t|d|dj dddt|d |d j dddnn|d}t|d|dj dddt|d |d j ddd|\}}t||jdd|j}|jdd di|j}t|j |dj ddt|j |dj dddt|tr|d}t|d|dj dddt|d |d j ddd|d d}t|d|dj dddt|d |d j dddnn|d}t|d|dj dddt|d |d j ddd|\}}t||jdd|j}|j|fdtjdd|j}t|j |dj d ddt|j |dj d ddd|}t|dddf|dj d dddt|ddd f|d j d ddd|\}}t||jdd|j}|jddtjddd|j}t|j |dj d ddt|j |dj d ddd|}t|dddf|dj d dddt|ddd f|d j d ddd|\}}t||jdd||j \}}t)|dtdd} tjt-|jj} d | dd<| | z} |jd| ddd|j} | } |dd| } | }t| j | j dt|| ddS) Nrb-C6?rse)ratolllgMbP?ruldeltamethodaverageTga2U0*3?g~jth?gMb@?probwhichy_valuesr<r=r8g{Gzt?)valuemean)exogr<r8)r<r8 agg_weightsvIh%<=gHz>)rrr#r$rCrBresults_margins_atmeansget_predictionpred_kwds_meanr predictedilocr1 isinstancerconf_intt_testtvaluesresults_margins_meanarangerzeroslenmodelendog pred_kwds_6 summary_frame)r&rrexrdfpredcistat_df6awpm6dfm6pmwdfmws r+ test_predictzCheckPredict.test_predict;s3yy Z   " "1 % %*"t"2==)<==C a(8tDDDDT!2TJJJJ d6 7 7 M#B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L Lg..q1B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L L L#B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L L++--adlTBBBB'"t"GG4G43FGGC a(8tDDDDT!2TJJJJ d6 7 7 M#B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L Lg..q1B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L L L#B BqE3t9>!#44t L L L L BqE3t9>!#44t L L L L++--adlTBBBB*"t"2:Vbill::%)%8::C ac(:FFFFT!!44tLLLL ]]__111a4#d).1"5DMMMM111a4#d).1"5DMMMM++--adlTBBBB'"t"H")A,,+/HH373FHHC ac(:FFFFT!!44tLLLL ]]__111a4#d).1"5DMMMM111a4#d).1"5DMMMM++--adlTBBBB++DN+33aT12A2h Xc$**++ , ,2A2 bggii!d!6s&$66$($466  ""!!"!MM  "" s}5AAAAd......r-c`|j}|d}|tjd}t |jjddt |jjdd | d dS#t$rYdSwxYw) N)y_max) bin_edgesrr>)upp_xlim) rget_diagnostictest_chisquare_probr#rQrdiff1shapeprobs_predicted plot_probs ImportError)r&rdiares_chi2s r+test_diagnosticzCheckPredict.test_diagnosticsy!!!++**RYq\\*BBX^)!,a000S(.q12666  NNBN ' ' ' ' '    DD sB B-,B-N)__name__ __module__ __qualname__r,rdrvrFr-r+rr-sH J J JX/X/X/t     r-rc&eZdZdZdZdZdZdS) CheckExtrasc|j}tjtdd}|j|fddi|j}t |j|jdz }|dkr5tj |tj |jd|ff}| |}| }| }t||ddS)Nr<linearrr-q=r)rr#r$rCrHrIrSr!rp column_stackrRrNrMr)r&rrXrZk_extrattcipcits r+test_predict_linearzCheckExtras.test_predict_linearsy ZRaR ! !"t"2MMXM9LMMdk""RXa[0 Q;;"bh W/E&F&F!GHHB [[__mmookkmmSu------r-c|j}|ttjddddf}|dd}t j|jdddg|jddg}t j dt|f}d|d<|j jddddf}ddl m }|||d }|||df } t| dd |dd |||d|d } |||dfd } t| dd | dd dS)Nrnewton,r7maxiterrr)r) score_test)params_constrained k_constraints exog_extrar:HC0)rrr_matrixcov_typerr)rklassrUrCvaluesfitr# concatenater!rRrSrT%statsmodels.base._parameter_inferencerr) r&rmodrresr params_restrrrrsc1sc2sc1_hcsc2_hcs r+test_score_testzCheckExtras.test_score_testsyzz%QQQV!455xxxx55~t{3B3'7!'+{233'7'9:: 8QL 1 1233Z_QQQV, DDDDDDj,'(***j:t*<===BQBRaR)))D\*+hPPPDj$-?%*,,,rr F2A2J/////r-c|ttjddddf}|dd}t j|jdddgg}t jdt|f}d|d<t j dd t j tj dz}dd lm}m}||d|d }t j|drJ||d|f } | dd ksJ||d|fd } | dd ksJdS)Nrrrrrr)rriW:g{Gz?)r_scorehess_extra)r exog2_extra hess_kwdsrrr)rrUrCrrr#rr!rRrSrandomseedrandnrprrrisnanany) r&rrrrrrrshrrs r+test_score_test_alphaz!CheckExtras.test_score_test_alphaszz%QQQV!455xxxx55~t{111~s&;<< 8QL 1 1233 y!!!BIOOEKN;;;  * * * * * * * *  dt*4FFF8BqE??&&(((((j4*<===1v}}}}DdJ-?%*,,,ay4r-c|j}ddlm}||}gd}|D]}t|||dS)Nr) MLEInfluence)cooks_distanced_fittedvaluesd_fittedvalues_scaledd_paramsdfbetashat_matrix_diagresid_studentized)r$statsmodels.stats.outliers_influencergetattrrW)r&rrinfluattrsattrs r+test_influencezCheckExtras.test_influenceswyEEEEEE T"" ! !D E4  r-N)rwrxryrrrrrFr-r+r{r{sQ . . .000,    D     r-r{c$eZdZedZdS)TestNegativeBinomialPPredictct|_tttdd}||_t j|_i|_ i|_ d|_ d|_ dS)Nrrrr:0yE>) r rrUrCrrrespresults_nb_docvisrrIrVr r)clsrs r+ setup_classz(TestNegativeBinomialPPredict.setup_classs]%  --1131OO) r-Nrwrxry classmethodrrFr-r+rr-  [   r-rc$eZdZedZdS)TestZINegativeBinomialPPredictczttdd}ttt|d}t jgd}||dd }||_tj |_ d | d i|_ d |dd i|_d|_d |_dS)NagetF)prependr:) exog_inflp) gRQg{GzgRQ?g{Gz?g)\(?ggQ?g(\?gHzG?g(\?gףp= ?rr start_paramsr7rrrrAr0)rrrrUrCr#arrayrrrresults_zinb_docvisrrBrIrVr r)rrmod_zinbsprs r+rz*TestZINegativeBinomialPPredict.setup_class s fu=== 0;DKKKX ||Hc|JJ+)9>>!+<+<=& "1" 6 r-NrrFr-r+rr s-[r-rc$eZdZedZdS)TestGeneralizedPoissonPredictct|_ttt}|d}||_t j|_i|_ i|_ d|_ d|_ dS)Nrr6rr) r rrUrCrrrrrrIrVr r)rmod1rs r+rz)TestGeneralizedPoissonPredict.setup_class!s]& !%..xxxx(() r-NrrFr-r+rrrr-r?gg?rr:loglike_methodnb1 geometricctjdd}tj|df}tj|}dd|zdzz}tj|dz}||fS)N\c:rrr:?)r#rronesrtrunc)nobsxynys r+get_data_simulatedrOsmINN9 D q A   B S2XMA 3A a4Kr-casec tt}}t|}tjd|\}}}t |tr|dkt}|||fi|}|}| |} | dj dksJ | |df} n8#t$r+| |} YnwxYw|| |fi|}||dd} | | j} t%| d| d t%| d| d | d } t%| | d | }t%| j| |z tj| z d t |ts| dtjd}|jt|jdfksJ| dtjdd}t%|j|dd | }|j|t>vrtAj!5tAj"dtF| $}|%dddn #1swxYwY|j&jt| fksJ |'}|jt| fksJn#tP$rYnwxYw|)}|jt| fksJ|j*}|jt| fksJ tAj!5tAj"dtF|+ddddS#1swxYwYdS#tX$rYdSwxYwdS)Nrrr)sizebfgsrrrg?rvar)r<rrEr9r}r;Tr?g|=ignore)category)-y_countx_constrSr#rr issubclassrastypefloatget_distributionpmfndimrvssqueeze ValueErrorrr!rrBrpredict resid_pearsonsqrtrQrprUrHrJrmrq models_influwarningscatch_warnings simplefilter UserWarning get_influencerWresidresid_score_factorAttributeError resid_scorerplot_influencers)rrrr cls_modelkwdsr!mod params_dgpdistry2resdistr2var_rBprobsprobs2rtrrf_scs r+ test_distrr\s, GqA q66DINN9"ItV)[))$ W  U # # )Aq ! !D ! !CJ   , ,E 99Q<<  ! ! ! !, YYT1IY & & . . 0 0 ,,, YYDY ! ! ) ) + +, )B " "T " "C ''z&#' F FC ! !#* - -FFKKMM))++A. DDDDFJJLL((**1-rvvxxcBBBB ;;U; # #DD&**,,..00u==== ;;==DC%T RWT]]'BOOOO i - - &29Q<< @@{s39~~q11111##29Q<<$??(%**Q--eDDDD  "" L   $ & & " "  !([ A A A A%%''E    ! ! !  " " " " " " " " " " " " " " " { SWWK//// ,,..E;3r77+-----    D !!##{s2wwk))))*zc"gg[(((( (** ' '%hEEEE$$&&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' '    DD 5! sm8*C##2DD-AN==OO)/P P&%P&;S0S > S SSSS S&%S&)5__doc__rnumpyr# numpy.testingrrpyteststatsmodels.tools.toolsr&statsmodels.base._prediction_inferencer#statsmodels.discrete.discrete_modelrrr r r r r statsmodels.discrete.count_modelrrr5statsmodels.sandbox.regression.tests.test_gmm_poissonrresultsrrsplit XLISTEXOG2 endog_name exog_namesrUrCrr{rrrlogmualphawrmodelsrrrrmark parametrizerrFr-r+r)s77777777 000000MMMMMM GFFFFF,,,,,,/ 4 4 6 6   % % ' '* 4y @  Z JttttttttnSSSSSSSSl     <   \,     K    RVC[[  B#  R3%!"hbh2w//0#R1b%.)A)AB#c1XxrxB/G/GH"B!R(@(@A"S!Hhbh2u~.F.FG b("(B4..!HBHb%[112a("(B;"7"78XRXr5k223#q828RK#8#89r828RK001(%0("(B;2G2GH(+6"G &   "!  &%''((GG)(GGGr-