B Z_@sddlZddlmZddlmZdddZdd d Zed krd d dgZ d ddgZ dZ dZ e rxdD]Z ddlZeZede d\ZZZededeeede de edeeedeeeedeeedeeedqfWdS) N)arma_generate_sample)Arma ?c Cs|dkrdddg}|dkr$dddg}g}g}xt|D]}t|||d|| d}||8}t|} | d\} } } } }|| | | }tt |||}| dk r||tt | q6|tj t | q6Wtj |d d|d dft|t|fS) zrun Monte Carlo for ARMA(2,2) DGP parameters currently hard coded also sample size `nsample` was not a self contained function, used instances from outer scope now corrected Ng?ggg333333?g?i)r)rangermeanrZfitappendZ geterrorsnpsqrtdotZdiagnanZ zeros_likeZr_Zarray)niternsamplearmasigZresultsZ results_bse_Zy2Zarest2Zrhohat2aZcov_x2aZinfodictZmesgZierZerr2aZsige2arBlib/python3.7/site-packages/statsmodels/sandbox/mcevaluate/arma.pymcarma22 s$      rcCs|dkrt|jd}t|d}td|||}tdtt||ddtdt||dtdtt||dtdtt|||d dtd ttt ||dtd t||kddS) Nrz!fractions of iterations with nansZRMSErrz mean biasz median biaszmedian bias percentdzmedian absolute errorzpositive error fraction) r ZzerosshapeZisnananyprintr r Zmedianabs)ZresrtZnanrowsrrr mc_summary-s" r__main__g?ggg333333?g?T)g?g?g?r)rrz Results for Monte Carlotruez nsample =zsigma = zelapsed time for Monte Carloz MC of rho versus truez MC of bse versus zeroz MC of bse versus std)rrNNr)N)Znumpyr Zstatsmodels.tsa.arima_processrZstatsmodels.tsa.arma_mlerrr__name__rrrZrun_mcrZtimeZt0rZres_rhoZres_bserZstdrrrrs2   $  8