!g<dZddlZddlZddlmZmZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZddlmZmZmZddlmZd d lmZd j;Zd Zd j;ezdgzZ eeZ!ee Z"GddZ#GddZ$Gdde#e$Z%Gdde#Z&Gdde$Z'ejPdZ)dZ*dZ+eiejXdgfeiejXdgfeiejXe+e)gfeiejXe+e)e*gfedd iejXe+e)e*gfeiejXe+e)e*gfedd iejXe+e)e*gfeiejXe)gfeiejXe)e*gfedd iejXe)e*gfeiejXe)e*gfeddiejXe)e*gfeiejXe)e*gfeddiejXe)e*gfedd iejXe)gfgZ-eeeeeeeegZ.d!Z/e/\Z0Z1ejdjgd"e-d#Z4y)$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 medicaidconstceZdZdZdZdZy) CheckPredictcD|j}|j}t|jdd}td|jdz d}t |j ||j ||j t |j||j|d|j ztj|j }tj|j }t |dtj|d|j y)Nrrrtol) res1res2slicek_inflrparamsrbsenpasarrayexp)selfrrsl1sl2params1params2s d/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/discrete/tests/test_predict.py test_basiczCheckPredict.test_basic/syyyyDKKT*Aq)40 C($++c*:K txx}2 >J**T[[)**T[[) RVVGBK%8tyyIc |j}|j}tjtj d}|j }|j|fi|j}t|j|djddt|j|djdddt|tr|jd}t|d|djdddt|d |d jddd|jd d}t|d|djdddt|d |d jdddnW|jd}t|d|djdddt|d |d jddd|j!\}}t||j"dd|j$}|jdd di|j}t|j|djddt|j|djdddt|tr|jd}t|d|djdddt|d |d jddd|jd d}t|d|djdddt|d |d jdddnW|jd}t|d|djdddt|d |d jddd|j!\}}t||j"dd|j }|j|fdtj&dd|j}t|j|djd ddt|j|djd ddd|j}t|dddf|djd dddt|ddd f|d jd ddd|j!\}}t||j"dd|j$}|jddtj&ddd|j}t|j|djd ddt|j|djd ddd|j}t|dddf|djd dddt|ddd f|d jd ddd|j!\}}t||j"dd|j!|j\}}t)|dtdd} tj*t-|j.j0} d | dd| | j z} |jd| ddd|j2} | j5} |jdd| } | j5}t| j| jdt|| dy) 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;syyyy ZZ  " "1 %**"t""2=)<)<=C a(8tDT!2TJ d6 7#B BqE3t9>>!#44t L BqE3t9>>!#44t Lg.q1B BqE3t9>>!#44t L BqE3t9>>!#44t L#B BqE3t9>>!#44t L BqE3t9>>!#44t L++-adllTB''"t""G4G43F3FGC a(8tDT!2TJ d6 7#B BqE3t9>>!#44t L BqE3t9>>!#44t Lg.q1B BqE3t9>>!#44t L BqE3t9>>!#44t L#B BqE3t9>>!#44t L BqE3t9>>!#44t L++-adllTB**"t""2:Vbiil:%)%8%8:C a(:FT!!44tL ]]_1a4#d)..1"5DM1a4#d)..1"5DM++-adllTB''"t""H"))A,+/H373F3FHC a(:FT!!44tL ]]_1a4#d)..1"5DM1a4#d)..1"5DM++-adllTB++DNN+3aT12Ah XXc$****+ ,2A bggi!d!!6s&$6$($4$46  "!!"!M  " s}}5Ad.r-cZ|j}|jd}|jtjd}t |j jddt |jjdd |jd y#t$rYywxYw) N)y_max) bin_edgesrr>)upp_xlim) rget_diagnostictest_chisquare_probr#rQrdiff1shapeprobs_predicted plot_probs ImportError)r&rdiares_chi2s r+test_diagnosticzCheckPredict.test_diagnosticsyy!!!+**RYYq\*BX^^))!,a0S((..q126  NNBN '   s B B*)B*N)__name__ __module__ __qualname__r,rdrvrFr-r+rr-s JX/t r-rc$eZdZdZdZdZdZy) CheckExtrasc|j}tjtdd}|j|fddi|j }t |j|jdz }|dkDr9tj|tj|jd|ff}|j|}|j}|j}t||dy)Nr<linearrr-q=r)rr#r$rCrHrIrSr!rp column_stackrRrNrMr)r&rrXrZk_extrattcipcits r+test_predict_linearzCheckExtras.test_predict_linearsyy ZZRa !"t""2MXM9L9LMdkk"RXXa[0 Q;"bhh W/E&F!GHB [[_mmokkmSu-r-c@|j}|jttjddddf}|j dd}t j|jdddg|jddg}t jdt|f}d|d<|jjddddf}ddl m }|||d }|||df } t| dd |dd |||d|d } |||dfd } t| dd | dd y)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_tests&yyzz%QV!45xxxx5~~t{{3B'7!'+{{23'7'9: 88QL 123ZZ__QV, D,'(*:t*<=BQRa)D\*+hPDj$-?%*,r F2AJ/r-c|jttjddddf}|j dd}t j |jdddgg}t jdt|f}d|d<t jjdd t jjtjdz}dd lm}m}||d|d }t j"|dj%rJ||d|f } | dd kDsJ||d|fd } | dd kDsJy)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_alphas,zz%QV!45xxxx5~~t{{1~s&;< 88QL 123 y!BIIOOEKKN;;  * dt*4F88BqE?&&(((4*<=1v}}DdJ-?%*,ay4r-c|j}ddlm}||}gd}|D]}t|||j y)Nr) MLEInfluence)cooks_distanced_fittedvaluesd_fittedvalues_scaledd_paramsdfbetashat_matrix_diagresid_studentized)r$statsmodels.stats.outliers_influencergetattrrW)r&rrinfluattrsattrs r+test_influencezCheckExtras.test_influencesHyyET" !D E4  ! r-N)rwrxryrrrrrFr-r+r{r{s .0,  D r-r{ceZdZedZy)TestNegativeBinomialPPredictct|_tttj dd}||_t j|_i|_ i|_ d|_ d|_ y)Nrrrr:0yE>) r rrUrCrrrespresults_nb_docvisrrIrVr r)clsrs r+ setup_classz(TestNegativeBinomialPPredict.setup_classsV%  -1131O)) r-Nrwrxry classmethodrrFr-r+rr  r-rceZdZedZy)TestZINegativeBinomialPPredictcLttdd}ttt|d}t j gd}|j|dd }||_tj|_ d |jd i|_ d |dd i|_d|_d |_y)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;DKXX ||Hc|J++)9>>!+<=& "1 6 r-NrrFr-r+rr sr-rceZdZedZy)TestGeneralizedPoissonPredictct|_ttt}|j d}||_t j|_i|_ i|_ d|_ d|_ y)Nrr6rr) r rrUrCrrrrrrIrVr r)rmod1rs r+rz)TestGeneralizedPoissonPredict.setup_class!sV& !%.xxxx()) r-NrrFr-r+rrrr-r?gg?rr:loglike_methodnb1 geometricctjjdd}tj|df}tjj |}dd|zdzz}tj |dz}||fS)N\c:rrr:?)r#rronesrtrunc)nobsxynys r+get_data_simulatedrOsgIINN9 D q A  B S2XMA 3A a4Kr-casectt}}t|}tjj d|\}}}t |tr|dkDjt}|||fi|}|}|j|} | jdjdk(sJ | j|dfj} || |fi|}|j!|dd} |j| j"} t%| j'jd| j'd t%| j)jd| j)d | j+d } t%| | j)jd | j+}t%| j,| |z tj.| z d t |ts| j+dtj0d}|j2t|j4dfk(sJ| j7dtj0dd}t%|j8|j'dd | j;}|j<|t>vrFtAjB5tAjDdtF| jI}|jKdddjLj2t| fk(sJ |jO}|j2t| fk(sJ |jS}|j2t| fk(sJ|jT}|j2t| fk(sJ tAjB5tAjDdtF|jWdddyy#t$r$| j|j} YnwxYw#1swY)xYw#tP$rYwxYw#1swYyxYw#tX$rYywxYw)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 q6DIINN9"ItV)[) W  U # Aq !D !CJ   ,E 99Q<   !! !, YYT1IY & . . 0 B "T "C ''z&#' FC ! !#** -FFKKM))+A. DFJJL((*1-rvvxcB ;;U; #DD&**,..0u= ;;=DC%%T RWWT]'BO i - &299Q< @{{s399~q1111##299Q<$?((%**Q-eD  " L  $ $ & "  ! !([ A%%'E    !  " {{  SWK/// ,,.E;;3r7+- --!!#{{s2wk)))**zzc"g[((( ((* '%%hE$$& ' '-!9 , YYDY ! ) ) +,: " "    ' '   s`#"P+Q=Q,Q=Q)P43P47Q QQQQQ Q+*Q+)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)s7 0M G,/ 4 4 6   % % '* 4y @  Z JttnSSl < \, K  RVVC[  B#  R3%!"hbhh2w/0#R1b%.)AB#c1XxrxxB/GH"B!R(@A"S!Hhbhh2u~.FG b("((B4.!HBHHb%[12a("((B;"78XRXXr5k23#q8288RK#89r8288RK01(%0("((B;2GH(+6"G &   "!  &'(G)Gr-